edits

xai and stuff

Author: rcghpge

Vision/Prototype.ipynb

@@ -476,16 +476,16 @@
     }
    ],
    "source": [
-    "# ---------------------------------------------\n",
+    "\n",
     "# Check number of samples per split\n",
-    "# ---------------------------------------------\n",
+    "\n",
     "print(\"\\nDataset sample counts per split:\")\n",
     "for split_name in dataset_dict:\n",
     "    print(f\"{split_name}: {len(dataset_dict[split_name])} samples\")\n",
     "\n",
-    "# ---------------------------------------------\n",
+    "\n",
     "# Print keys of first example\n",
-    "# ---------------------------------------------\n",
+    "\n",
     "example = dataset_dict[\"train\"][0]\n",
     "print(\"\\nFirst example keys:\", list(example.keys()))\n",
     "\n",
@@ -704,9 +704,9 @@
     "        ax.plot(stroke[:, 0], -stroke[:, 1], color=color, linewidth=2)\n",
     "    ax.axis(\"off\")\n",
     "\n",
-    "# From your dataloader, get raw traces instead of padded\n",
+    "# From the dataloader, the model can pull raw traces instead of padded\n",
     "for batch in tf_train.take(1):\n",
-    "    # This assumes you still have original \"traces\" column in dataset_dict\n",
+    "    # This is if you still have original \"traces\" in a dataset_dict\n",
     "    sample_idx = 0\n",
     "    sample_file_path = dataset_dict[\"train\"][sample_idx][\"file_path\"]\n",
     "    sample_label = dataset_dict[\"train\"][sample_idx][\"normalized_label\"]\n",