Revert testing change that broke example

brian-team · Jul 13, 2021 · 9545887 · 9545887
1 parent 7461b39
commit 9545887
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/IF_sbi.py b/examples/IF_sbi.py
@@ -57,9 +57,9 @@ def n_peaks(x):
 inferencer = Inferencer(dt=dt, model=eqs_inf,
                         input={'I_syn': inp_trace.reshape(1, -1)},
                         output={'v': out_trace},
-                        features=[lambda x: x[(t > start_syn) & (t < end_syn), :].mean(axis=1),
-                                  lambda x: x[(t > start_syn) & (t < end_syn), :].std(axis=1),
-                                  lambda x: x[(t > start_syn) & (t < end_syn), :].max(axis=1),
+                        features=[lambda x: x[(t > start_syn) & (t < end_syn), :].mean(axis=0),
+                                  lambda x: x[(t > start_syn) & (t < end_syn), :].std(axis=0),
+                                  lambda x: x[(t > start_syn) & (t < end_syn), :].max(axis=0),
                                   n_peaks],
                         method='exponential_euler',
                         threshold='v > -50 * mV',