Add confusion matrix calculation

Author: abdullahkady

main.ipynb

@@ -80,7 +80,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Labels matrix (), each row representing one class, and the columns represent the true label of each image \n",
+    "# Labels matrix, each row representing one class, and the columns represent whether\n",
+    "# the image belongs to that class (row index) or not (-1, 1) \n",
     "T = np.zeros((10, 2400))\n",
     "for i in range(len(T)):\n",
     "    T[i] = np.where(t == i, 1, -1)\n",
@@ -117,8 +118,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "out = np.dot(x_delta, WC)\n",
-    "WC.shape"
+    "predictions = np.dot(x_delta, weights_matrix)\n",
+    "conf_matrix = np.zeros((10, 10))\n",
+    "for i, row in enumerate(predictions):\n",
+    "    prediction = np.where(row == row.max())[0][0] # The index (class) of the maximum value (most probable class)\n",
+    "    true_value = t[i]\n",
+    "    conf_matrix[true_value][prediction] += 1\n",
+    "    \n",
+    "conf_matrix"
    ]
   },
   {
@@ -127,21 +134,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "tested = out[1000]\n",
-    "pred = np.where(zero == zero.max())\n",
+    "to_be_tested = predictions[2030]\n",
+    "pred = np.where(to_be_tested == to_be_tested.max())\n",
     "pred, pred[0][0]"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "scrolled": false
-   },
-   "outputs": [],
-   "source": [
-    "out.shape"
-   ]
   }
  ],
  "metadata": {