diff options
Diffstat (limited to 'experiment/analysis/tools.py')
| -rw-r--r-- | experiment/analysis/tools.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/experiment/analysis/tools.py b/experiment/analysis/tools.py index 1dffc9a..cde322f 100644 --- a/experiment/analysis/tools.py +++ b/experiment/analysis/tools.py @@ -44,8 +44,8 @@ def blocked_time(vp): result = {} sum_time = 0 - block_i = 0 - for trial in range(trial_count): + block_i = 1 + for trial in range(1, trial_count): if trial % 5 == 0: sum_time = 0 block_i += 1 @@ -119,7 +119,7 @@ def train_test_split(data): for vp in data[cond].keys(): new_dict[cond][vp] = {} for trial in data[cond][vp].keys(): - if string in trial: + if string in trial and trial != "train_0": new_dict[cond][vp][trial] = data[cond][vp][trial] return new_dict |