Commit 9a088f2f authored by Andrei Roibu's avatar Andrei Roibu
Browse files

fixed saving function

parent f5d7e08b
......@@ -97,8 +97,6 @@ def database_generator(data_directory, train_inputs, train_targets, rsfMRI_mean_
subject_t0 = datetime.now()
print("Processing subject {}/{}".format(index, number_of_subjects))
w_dMRI, w_rsfMRI = regression_weight_calculator(data_directory, subject, train_inputs, train_targets, rsfMRI_mean_mask_path, dMRI_mean_mask_path)
# ------------------
......@@ -127,15 +125,11 @@ def database_generator(data_directory, train_inputs, train_targets, rsfMRI_mean_
# ------------------
print("Processed subject {}/{} | Total Duration: {}".format(index, number_of_subjects, datetime.now() - subject_t0))
dmri_imaging_df = pd.DataFrame.from_dict(dmri_imaging_dictionary, orient="index", columns = dictionary_labels)
dmri_imaging_df.to_pickle('dmri_stats.pkl')
del dmri_imaging_df
dmri_imaging_df.to_pickle('utils/dmri_stats.pkl')
rsfmri_imaging_df = pd.DataFrame.from_dict(rsfmri_imaging_dictionary, orient="index", columns = dictionary_labels)
rsfmri_imaging_df.to_pickle('rsfmri_stats.pkl')
del rsfmri_imaging_df
rsfmri_imaging_df.to_pickle('utils/rsfmri_stats.pkl')
if __name__ == '__main__':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment