multi-network

aramis-lab · Oct 1, 2024 · 6533084 · 6533084
1 parent 06b8f8a
commit 6533084
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/clinicadl/trainer/tasks_utils.py b/clinicadl/trainer/tasks_utils.py
@@ -583,6 +583,7 @@ def check_prediction(row):
         print(subject_df)
         for i in range(n_classes):
             print(subject_df[f"proba{i}"])
+        print(weight_series)
         proba_list = [
             np.average(subject_df[f"proba{i}"], weights=weight_series)
             for i in range(n_classes)