Commit ea89c4db authored by Andrei Roibu's avatar Andrei Roibu
Browse files

changed output stats type to csv

parent 282671ca
......@@ -2,7 +2,7 @@
trained_model_path = "saved_models/VA2-1.pth.tar"
prediction_output_path = "VA2-1_predictions"
prediction_output_database_name = "output_test_data.h5"
prediction_output_statistics_name = "output_statistics.pkl"
prediction_output_statistics_name = "output_statistics.csv"
data_directory = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/"
mapping_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
mapping_targets_file = "fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
......
......@@ -222,7 +222,7 @@ def evaluate_data(trained_model_path,
raise(exception_expression)
output_statistics_df = pd.DataFrame.from_dict(output_statistics, orient='index', columns=['mse', 'mae', 'cel', 'pearson_r', 'pearson_p', 'spearman_r', 'spearman_p', 'regression_w', 'regression_b'])
output_statistics_df.to_pickle(output_statistics_path)
output_statistics_df.to_csv(output_statistics_path)
log.info("Output Data Generation Complete")
......
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