Skip to content

Commit

Permalink
fix(Evaluation): specified groupby column to be consistent with panda…
Browse files Browse the repository at this point in the history
…s 2.X
  • Loading branch information
muellerdo committed Feb 21, 2024
1 parent c48c4c9 commit 8cb896f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aucmedi/evaluation/comparison.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,7 @@ def evaluate_comparison(pred_list,

# Optional: Macro average classes
if macro_average_classes:
metrics_avg = metrics.groupby(["metric", "model"]).mean()
metrics_avg = metrics.groupby(["metric", "model"])[["score"]].mean()
metrics = metrics_avg.reset_index()

# Append to dataframe list
Expand Down

0 comments on commit 8cb896f

Please sign in to comment.