remove unneeded prints

wasserth · Oct 4, 2024 · b33449b · b33449b
1 parent 7470a0f
commit b33449b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/totalsegmentator/bin/totalseg_get_modality.py b/totalsegmentator/bin/totalseg_get_modality.py
@@ -48,8 +48,6 @@ def get_modality(img: nib.Nifti1Image):
     for fold, clf in clfs.items():
         preds.append(clf.predict([features])[0])
     preds = np.array(preds)
-    print("Ensemble preds:")
-    print(preds)
     preds = np.mean(preds)
     prediction_str = "ct" if preds < 0.5 else "mr"
     probability = 1 - preds if preds < 0.5 else preds