From 3fa2b8f6b680818c183ad72b9770d90f56e39607 Mon Sep 17 00:00:00 2001
From: meganjkurka <20548431+meganjkurka@users.noreply.github.com>
Date: Fri, 23 Mar 2018 12:26:56 -0700
Subject: [PATCH] small update to glm demo (#509)
---
examples/py/demos/H2O4GPU_GLM.ipynb | 32 ++++++++++++++++++++---------
1 file changed, 22 insertions(+), 10 deletions(-)
diff --git a/examples/py/demos/H2O4GPU_GLM.ipynb b/examples/py/demos/H2O4GPU_GLM.ipynb
index 4cfe93388..3b6ae7326 100644
--- a/examples/py/demos/H2O4GPU_GLM.ipynb
+++ b/examples/py/demos/H2O4GPU_GLM.ipynb
@@ -3,7 +3,9 @@
{
"cell_type": "code",
"execution_count": 1,
- "metadata": {},
+ "metadata": {
+ "collapsed": true
+ },
"outputs": [],
"source": [
"import h2o4gpu\n",
@@ -16,7 +18,9 @@
{
"cell_type": "code",
"execution_count": 2,
- "metadata": {},
+ "metadata": {
+ "collapsed": false
+ },
"outputs": [
{
"name": "stdout",
@@ -77,7 +81,9 @@
{
"cell_type": "code",
"execution_count": 3,
- "metadata": {},
+ "metadata": {
+ "collapsed": false
+ },
"outputs": [
{
"name": "stdout",
@@ -100,7 +106,9 @@
{
"cell_type": "code",
"execution_count": 4,
- "metadata": {},
+ "metadata": {
+ "collapsed": false
+ },
"outputs": [
{
"name": "stdout",
@@ -135,7 +143,9 @@
{
"cell_type": "code",
"execution_count": 5,
- "metadata": {},
+ "metadata": {
+ "collapsed": false
+ },
"outputs": [
{
"name": "stdout",
@@ -160,14 +170,16 @@
"Make predictions on validation set\n",
"\"\"\"\n",
"print(\"Predictions per alpha\")\n",
- "preds = model.predict(valid_x, valid_y)\n",
+ "preds = model.predict_proba(valid_x, valid_y)\n",
"print(preds)"
]
},
{
"cell_type": "code",
"execution_count": 6,
- "metadata": {},
+ "metadata": {
+ "collapsed": false
+ },
"outputs": [
{
"name": "stdout",
@@ -202,9 +214,9 @@
],
"metadata": {
"kernelspec": {
- "display_name": "env",
+ "display_name": "Python 3",
"language": "python",
- "name": "env"
+ "name": "python3"
},
"language_info": {
"codemirror_mode": {
@@ -216,7 +228,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
- "version": "3.6.2"
+ "version": "3.6.0"
}
},
"nbformat": 4,