Skip to content

Commit

Permalink
minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
OnnoKampman committed Apr 28, 2024
1 parent 94b22b2 commit 19dfad8
Showing 1 changed file with 11 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -29,18 +29,23 @@
if not os.path.exists(kernel_params_savedir):
os.makedirs(kernel_params_savedir)

n_subjects = cfg['n-subjects']
num_subjects = cfg['n-subjects']
all_subjects_list = get_human_connectome_project_subjects(
data_dir=cfg['data-dir'], first_n_subjects=n_subjects, as_ints=True
data_dir=cfg['data-dir'],
first_n_subjects=num_subjects,
as_ints=True,
)

# TODO: refactor so we only load each model once?
for kernel_param in cfg['kernel-params']:
kernel_params_df = pd.DataFrame(index=all_subjects_list, columns=cfg['scan-ids'])
kernel_params_df = pd.DataFrame(
index=all_subjects_list,
columns=cfg['scan-ids'],
)
print('\nKERNEL PARAM:', kernel_param, '\n')

for i_subject, subject in enumerate(all_subjects_list):
print(f'\n> SUBJECT {i_subject+1:d} / {n_subjects:d}: {subject:d}')
print(f'\n> SUBJECT {i_subject+1:d} / {num_subjects:d}: {subject:d}')

for scan_id in cfg['scan-ids']:
model_savedir = os.path.join(
Expand All @@ -52,8 +57,8 @@
kernel_params_df.loc[subject, scan_id] = m_dict[kernel_param]

kernel_params_df_filename = f'{kernel_param:s}_kernel_params.csv'
kernel_params_df.to_csv(
kernel_params_df.astype(float).to_csv(
os.path.join(kernel_params_savedir, kernel_params_df_filename),
float_format='%.3f'
float_format='%.3f',
)
logging.info(f"Saved kernel params table '{kernel_params_df_filename:s}' to '{kernel_params_savedir:s}'.")

0 comments on commit 19dfad8

Please sign in to comment.