update

Author: rasoulameri

cnn_architecture

@@ -10,7 +10,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "4534a191",
    "metadata": {},
    "outputs": [
@@ -46,7 +46,6 @@
     "\n",
     "# --- Visualize some samples ---\n",
     "%matplotlib inline\n",
-    "\n",
     "fig, axes = plt.subplots(2, 5, figsize=(6, 4))\n",
     "\n",
     "for ax in axes.flat:\n",
@@ -263,7 +262,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "id": "4a1f7733",
    "metadata": {},
    "outputs": [
@@ -308,7 +307,7 @@
     "from src.models.cnn_model import CNNModel\n",
     "\n",
     "model = CNNModel(num_classes=10)\n",
-    "summary(model, input_size=(1, 1, 28, 28))\n"
+    "summary(model, input_size=(1, 1, 28, 28))"
    ]
   },
   {
@@ -321,7 +320,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "id": "dcfe4491",
    "metadata": {},
    "outputs": [
@@ -353,41 +352,7 @@
     "    class_names=class_names,\n",
     "    max_samples=10,\n",
     "    layer_name=\"conv3\"\n",
-    ")\n"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "37ec9dab",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "\n",
-      "(process:23160): Pango-WARNING **: 18:20:07.026: couldn't load font \"Linux libertine Not-Rotated 10\", falling back to \"Sans Not-Rotated 10\", expect ugly output.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/plain": [
-       "'cnn_architecture.png'"
-      ]
-     },
-     "execution_count": 8,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "from torchview import draw_graph\n",
-    "from src.models.cnn_model import CNNModel\n",
-    "\n",
-    "model = CNNModel(num_classes=10)\n",
-    "graph = draw_graph(model, input_size=(1, 1, 28, 28))\n",
-    "graph.visual_graph.render(\"cnn_architecture\", format=\"png\")\n"
+    ")"
    ]
   }
  ],

cnn_architecture.png

@@ -1,5 +1,10 @@
 torch
 torchvision
 matplotlib
+seaborn
 numpy
-tqdm
+scikit-learn
+PyYAML
+opencv-python
+torchviz
+torchview

notebooks/EDA_and_Explainability.ipynb

@@ -35,7 +35,7 @@ def main():
         )
 
     os.makedirs(config["train"]["checkpoint_dir"], exist_ok=True)
-    best_path = os.path.join(config["train"]["checkpoint_dir"], "best_model.pth")
+    # best_path = os.path.join(config["train"]["checkpoint_dir"], "best_model.pth")
 
     train_model(
         model,