From 9545887a882183e3283260a3ee0c51748a0a7b95 Mon Sep 17 00:00:00 2001 From: Marcel Stimberg Date: Tue, 13 Jul 2021 15:59:15 +0200 Subject: [PATCH] Revert testing change that broke example --- examples/IF_sbi.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/IF_sbi.py b/examples/IF_sbi.py index b7b8ec0..8de5adc 100644 --- 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',