Update Diabetes Ridge Regression Training.ipynb

Author: GabbyHills

experimentation/Diabetes Ridge Regression Training.ipynb

@@ -46,6 +46,19 @@
     "    return data"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+   "# Train the model, return the model\n",
+   "def train_model(data, args):\n",
+   "reg_model = Ridge(**args)\n",
+   "reg_model.fit(data[\"train\"][\"X\"], data[\"train\"][\"y\"])\n",
+   "   return reg_model"
+   ]
+   },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -55,7 +68,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -88,10 +101,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [],
    "source": [
+   "# Train the model, return the model\n",
    "def train_model(data, args):\n",
    "reg_model = Ridge(**args)\n",
    "reg_model.fit(data[\"train\"][\"X\"], data[\"train\"][\"y\"])\n",