Skip to content

Commit 8c7541c

Browse files
authored
Merge pull request #34 from GabbyHills/GabbyHills-patch-38
Update Diabetes Ridge Regression Training.ipynb
2 parents f009f23 + fe808a2 commit 8c7541c

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

experimentation/Diabetes Ridge Regression Training.ipynb

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,18 @@
5959
" return reg_model"
6060
]
6161
},
62+
{
63+
"cell_type": "code",
64+
"metadata": {},
65+
"source": [
66+
"# Evaluate the metrics for the model\n",
67+
"def = get_model_metrics(reg_model, data):\n",
68+
"preds = reg_model.predict(data[\"test\"][\"X\"])\n",
69+
"mse = mean_squared_error(preds, y_test)\n",
70+
"metrics = {\"mse\": mse}\n",
71+
" return metrics"
72+
]
73+
},
6274
{
6375
"cell_type": "code",
6476
"execution_count": 4,
@@ -139,7 +151,7 @@
139151
]
140152
},
141153
{
142-
"cell_type": "markdown",
154+
"cell_type": "code",
143155
"metadata": {},
144156
"source": [
145157
"## Evaluate the metrics for the model"

0 commit comments

Comments
 (0)