From 36c60caa60d733dcf49711b1fbc134cb981b72c0 Mon Sep 17 00:00:00 2001 From: Onno Kampman Date: Wed, 8 May 2024 17:23:47 +0800 Subject: [PATCH] fix boxcar handling --- .../fmri/sim/train_models/save_TVFC_estimates.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/benchmarks/fmri/sim/train_models/save_TVFC_estimates.py b/benchmarks/fmri/sim/train_models/save_TVFC_estimates.py index 35488321..978101d5 100644 --- a/benchmarks/fmri/sim/train_models/save_TVFC_estimates.py +++ b/benchmarks/fmri/sim/train_models/save_TVFC_estimates.py @@ -43,10 +43,11 @@ cfg['data-dir'], noise_type, f'trial_{i_trial:03d}', 'checkerboard_covariance.csv' ) - if not os.path.exists(data_filepath): - logging.warning(f"Node time series not found: '{data_filepath:s}'") + if not os.path.exists(data_filepath): + logging.warning(f"Node time series not found: '{data_filepath:s}'") + continue + else: continue - continue tvfc_estimates_savedir = os.path.join( cfg['experiments-basedir'], noise_type, f'trial_{i_trial:03d}', 'TVFC_estimates', @@ -64,8 +65,6 @@ data_filepath, verbose=False, ) # (N, 1), (N, D) - num_time_steps = x.shape[0] - num_time_series = y.shape[1] match data_split: case "LEOO": @@ -86,7 +85,7 @@ data_split=data_split, noise_type=noise_type, i_trial=i_trial, - covs_type=covs_type + covs_type=covs_type, ) if estimated_tvfc is None: continue