Commit 72e9a9c9 authored by Andrei Roibu's avatar Andrei Roibu
Browse files

fixed bug where the path was passed to pkl not the df

parent 78f6ad04
......@@ -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_pickle(output_statistics)
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