Add test input for the lr model.

only4sim · Dec 26, 2023 · ae9c224 · ae9c224
1 parent 6cf2d39
commit ae9c224
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/src/.DS_Store b/src/.DS_Store
diff --git a/src/models/logistic_regression.ipynb b/src/models/logistic_regression.ipynb
@@ -56,6 +56,25 @@
     "print(intercept)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "[0]\n"
+     ]
+    }
+   ],
+   "source": [
+    "input = [2.5, 3.5]\n",
+    "output = clf.predict([input])\n",
+    "print(output)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},