cleanup + math typo

neuromatch · Jul 23, 2024 · 65b4d68 · 65b4d68
1 parent 49e589c
commit 65b4d68
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tutorials/W2D2_NeuroSymbolicMethods/W2D2_Tutorial1.ipynb b/tutorials/W2D2_NeuroSymbolicMethods/W2D2_Tutorial1.ipynb
@@ -1413,7 +1413,7 @@
     "        raise NotImplementedError(\"Student exercise: complete similarity calculation between input vector and weights of the network.\")\n",
     "        ###################################################################\n",
     "        sims = ...\n",
-    "        max_sim = softmax(sims * self.temp, axis=0)\n",
+    "        max_sim = softmax(sims * self.temp, axis=1)\n",
     "        return sspspace.SSP(...)\n",
     "\n",
     "\n",
@@ -1444,7 +1444,7 @@
     "        self.temp = temperature\n",
     "    def __call__(self, x):\n",
     "        sims = x @ self.weights.T\n",
-    "        max_sim = softmax(sims * self.temp, axis=0)\n",
+    "        max_sim = softmax(sims * self.temp, axis=1)\n",
     "        return sspspace.SSP(max_sim @ self.weights)\n",
     "\n",
     "\n",
@@ -2188,7 +2188,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.19"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,