Skip to content

Commit

Permalink
fix(Evaluation): changed datatype from float to object for ROC multi-…
Browse files Browse the repository at this point in the history
…lists
  • Loading branch information
muellerdo committed Feb 21, 2024
1 parent 812bb88 commit f6945b8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aucmedi/evaluation/performance.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,7 +236,7 @@ def evalby_barplot(metrics, out_path, class_names, show=False, suffix=None):
#-----------------------------------------------------#
def evalby_rocplot(fpr_list, tpr_list, out_path, class_names, show=False, suffix=None):
# Initialize result dataframe
df_roc = pd.DataFrame(data=[fpr_list, tpr_list], dtype=np.float64)
df_roc = pd.DataFrame(data=[fpr_list, tpr_list], dtype=np.object)
# Preprocess dataframe
df_roc = df_roc.transpose()
df_roc = df_roc.apply(pd.Series.explode)
Expand Down

0 comments on commit f6945b8

Please sign in to comment.