Skip to content

Commit b7719ff

Browse files
committed
Remove uselec 'precion' renamique in df
1 parent 3ef64b8 commit b7719ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/assess_detections.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ def main(cfg_file_path):
418418
tmp_df = metrics_by_cl_df[['dataset', 'TP_k', 'FP_k', 'FN_k']].groupby(by='dataset', as_index=False).sum()
419419
tmp_df2 = metrics_by_cl_df[['dataset', 'precision_k', 'recall_k']].groupby(by='dataset', as_index=False).mean()
420420
global_metrics_df = tmp_df.merge(tmp_df2, on='dataset')
421-
global_metrics_df.rename({'TP_k': 'TP', 'FP_k': 'FP', 'FN_k': 'FN', 'precision_k': 'precision', 'precision_k': 'precision', 'recall_k': 'recall'}, inplace=True)
421+
global_metrics_df.rename({'TP_k': 'TP', 'FP_k': 'FP', 'FN_k': 'FN', 'precision_k': 'precision', 'recall_k': 'recall'}, inplace=True)
422422

423423
file_to_write = os.path.join(OUTPUT_DIR, 'global_metrics.csv')
424424
global_metrics_df.to_csv(file_to_write, index=False)

0 commit comments

Comments
 (0)