Calculate the mean of each metric

UofT-DSI · Nov 10, 2024 · 8768463 · 8768463
1 parent 7900d4b
commit 8768463
Showing 1 changed file with 28 additions and 3 deletions.
diff --git a/02_activities/assignments/assignment_2.ipynb b/02_activities/assignments/assignment_2.ipynb
@@ -362,10 +362,35 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 51,
    "metadata": {},
-   "outputs": [],
-   "source": []
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Mean values for each metric across the folds:\n",
+      "fit_time                   3.106709\n",
+      "score_time                 0.152477\n",
+      "test_neg_log_loss         -0.328118\n",
+      "train_neg_log_loss        -0.311814\n",
+      "test_roc_auc               0.910637\n",
+      "train_roc_auc              0.925373\n",
+      "test_accuracy              0.856823\n",
+      "train_accuracy             0.865322\n",
+      "test_balanced_accuracy     0.750700\n",
+      "train_balanced_accuracy    0.763034\n",
+      "dtype: float64\n"
+     ]
+    }
+   ],
+   "source": [
+    "mean_metrics = cv_results_df.mean()\n",
+    "\n",
+    "# Displaying the mean values for each metric\n",
+    "print(\"Mean values for each metric across the folds:\")\n",
+    "print(mean_metrics)\n"
+   ]
   },
   {
    "cell_type": "markdown",