Update NMNIST.ipynb

SpikeAI · Dec 6, 2023 · 44779a9 · 44779a9
1 parent d723f37
commit 44779a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/dataloaders/NMNIST.ipynb b/examples/dataloaders/NMNIST.ipynb
@@ -194,11 +194,12 @@
         "\n",
         "# flatten on-spikes and off-spikes into one channel\n",
         "# a = (train_dl.dataset[n][0][:, 0] + train_dl.dataset[n][0][:, 1])\n",
+        "data = (data>=1).float() # some spikes are equal to 2...\n",
         "a = (data[:, 0, :, :] - data[:, 1, :, :])\n",
         "# a = np.swapaxes(a, 0, -1)\n",
         "#  Plot\n",
         "fig, ax = plt.subplots()\n",
-        "anim = splt.animator(a, fig, ax, interval=30)\n",
+        "anim = splt.animator(a, fig, ax, interval=30, cmap='seismic')\n",
         "HTML(anim.to_html5_video())\n",
         "# anim.save('nmnist_animation.mp4', writer = 'ffmpeg', fps=50)  "
       ]