Add neuron indices

Bench: 2134743
SzilBalazs · Aug 16, 2023 · f4ab06a · f4ab06a
1 parent c9da70e
commit f4ab06a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/network_to_img.py b/scripts/network_to_img.py
@@ -63,6 +63,9 @@ def main():
             for i in range(12):
                 axs[7][i].set_xlabel(col_labels[i])
 
+            for i in range(8):
+                axs[i][0].set_ylabel(i + base * 8 + 1)
+
             plt.savefig(f'img/network_{base}.png')