test

Author: DerikVo

Created_images/Baseline confusion matrix.png

@@ -53,7 +53,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 5,
    "id": "17b38552-4ffe-4ac9-b9ba-2ee96faa0e8d",
    "metadata": {
     "tags": []
@@ -68,14 +68,6 @@
      },
      "metadata": {},
      "output_type": "display_data"
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "          Accuracy  Precision   Recall  F1 Score\n",
-      "baseline   0.46987   0.468308  0.46987  0.463567\n"
-     ]
     }
    ],
    "source": [
@@ -86,19 +78,77 @@
     "cm, metrics_df = m.calculate_metrics(actual_classes, predicted_classes, class_paths)\n",
     "\n",
     "# Plot the confusion matrix\n",
-    "m.plot_confusion_matrix(cm, class_paths, 'Baseline')\n",
-    "\n",
-    "# Display the metrics DataFrame\n",
-    "print(metrics_df)"
+    "m.plot_confusion_matrix(cm, class_paths, 'Baseline')"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 4,
    "id": "ac7d7db5-1ca0-493a-8683-34660e799658",
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>Accuracy</th>\n",
+       "      <th>Precision</th>\n",
+       "      <th>Recall</th>\n",
+       "      <th>F1 Score</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>baseline</th>\n",
+       "      <td>0.46987</td>\n",
+       "      <td>0.468308</td>\n",
+       "      <td>0.46987</td>\n",
+       "      <td>0.463567</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "          Accuracy  Precision   Recall  F1 Score\n",
+       "baseline   0.46987   0.468308  0.46987  0.463567"
+      ]
+     },
+     "execution_count": 4,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "metrics_df"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "48a01f31-0afb-4cc5-9de3-3f0b4e4a3eba",
+   "metadata": {},
    "outputs": [],
-   "source": []
+   "source": [
+    "metrics_df.to_csv('../NoteBooks/scores.csv', index_label='Model')"
+   ]
   }
  ],
  "metadata": {