fix corrupted notebook

agosztolai · Jun 17, 2024 · 6183031 · 6183031
1 parent 0575515
commit 6183031
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/MARBLE/layers.py b/MARBLE/layers.py
@@ -87,7 +87,7 @@ def __init__(self, C, D):
 
         self.O_mat = nn.ModuleList()
         for _ in range(C):
-            self.O_mat.append(nn.Linear(D, D, bias=True))
+            self.O_mat.append(nn.Linear(D, D, bias=False))
 
         self.reset_parameters()
 

diff --git a/examples/RNN/RNN.ipynb b/examples/RNN/RNN.ipynb
@@ -1076,10 +1076,8 @@
     "        v = pos[j]\n",
     "\n",
     "        n_samples = np.min([len(u), len(v)])\n",
-    "        ind = np.random.choice(len(u), size=(n_samples,), replace=False)\n",
-    "        u = u[np.sort(ind)]\n",
-    "        ind = np.random.choice(len(v), size=(n_samples,), replace=False)\n",
-    "        v = v[np.sort(ind)]\n",
+    "        u = u[:n_samples]\n",
+    "        v = v[:n_samples]\n",
     "          \n",
     "        u_score, v_score = cca.fit_transform(u, v)\n",
     "\n",
@@ -1196,7 +1194,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },