bugfix

trieschlab · May 4, 2023 · 26941a5 · 26941a5
1 parent d727610
commit 26941a5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/NetworkBehavior/Basics/Normalization.py b/NetworkBehavior/Basics/Normalization.py
@@ -1,3 +1,4 @@
+from PymoNNto.NetworkCore.Neuron_Group import *
 from PymoNNto.NetworkCore.Behavior import *
 import numpy as np
 
@@ -35,12 +36,12 @@ def normalize_synapse_attr(src_attr, target_attr, target_value, neurons, synapse
 def normalize_synapse_attr_efferent(src_attr, target_attr, target_value, neurons, synapse_type):
     neurons.temp_weight_sum = neurons.vector()
 
-    for s in neurons.efferent_synapses[synapse_type]:
+    for s in neurons.synapses(efferent, synapse_type):
         s.src.temp_weight_sum += np.sum(np.abs(getattr(s, target_attr)), axis=0)
 
     neurons.temp_weight_sum /= target_value
 
-    for s in neurons.efferent_synapses[synapse_type]:
+    for s in neurons.neurons.synapses(efferent, synapse_type):
         setattr(s, src_attr, getattr(s, src_attr) / (s.src.temp_weight_sum + (s.src.temp_weight_sum == 0)))
 
 '''

diff --git a/NetworkBehavior/Basics/Normalization_Sparse.py b/NetworkBehavior/Basics/Normalization_Sparse.py
@@ -1,5 +1,7 @@
+from PymoNNto.NetworkCore.Neuron_Group import *
 import numpy as np
 
+
 def normalize_synapse_attr_sparse(src_attr, target_attr, target_value, neurons, synapse_type):
 
     neurons.temp_weight_sum = neurons.vector()

diff --git a/NetworkBehavior/Basics/Normalization_Tensorflow.py b/NetworkBehavior/Basics/Normalization_Tensorflow.py
@@ -1,3 +1,4 @@
+from PymoNNto.NetworkCore.Neuron_Group import *
 from PymoNNto.NetworkCore.Behavior import *
 import numpy as np
 import tensorflow as tf