Merge pull request #474 from petermarathas/fix_plot_spike_ylabels

fix yticks in plot_spikes
BindsNET · Apr 15, 2021 · d3eca0d · d3eca0d
2 parents 2eb3aa5 + 8c19102
commit d3eca0d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindsnet/analysis/plotting.py b/bindsnet/analysis/plotting.py
@@ -140,13 +140,13 @@ def plot_spikes(
             axes[i].set_title(
                 "%s spikes for neurons (%d - %d) from t = %d to %d " % args
             )
+            axes[i].set_yticks([n_neurons[datum[0]][0], n_neurons[datum[0]][1]])
         for ax in axes:
             ax.set_aspect("auto")
 
         plt.setp(
             axes,
             xticks=[],
-            yticks=[],
             xlabel="Simulation time",
             ylabel="Neuron index",
         )
@@ -174,6 +174,7 @@ def plot_spikes(
             axes[i].set_title(
                 "%s spikes for neurons (%d - %d) from t = %d to %d " % args
             )
+            axes[i].set_yticks([n_neurons[datum[0]][0], n_neurons[datum[0]][1]])
 
     plt.draw()