fix(Evaluation): changed datatype from float to object for ROC multi-…

…lists
frankkramer-lab · Feb 21, 2024 · f6945b8 · f6945b8
1 parent 812bb88
commit f6945b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aucmedi/evaluation/performance.py b/aucmedi/evaluation/performance.py
@@ -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)