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

changed output stats type to csv

parent 282671ca
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
trained_model_path = "saved_models/VA2-1.pth.tar" trained_model_path = "saved_models/VA2-1.pth.tar"
prediction_output_path = "VA2-1_predictions" prediction_output_path = "VA2-1_predictions"
prediction_output_database_name = "output_test_data.h5" 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/" data_directory = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/"
mapping_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz" mapping_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
mapping_targets_file = "fMRI/rfMRI_25.dr/dr_stage2.nii.gz" mapping_targets_file = "fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
......
...@@ -222,7 +222,7 @@ def evaluate_data(trained_model_path, ...@@ -222,7 +222,7 @@ def evaluate_data(trained_model_path,
raise(exception_expression) 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 = 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") 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