From b33449bc84c85a490b79f0d0fd2e4dd464ef00ed Mon Sep 17 00:00:00 2001 From: wasserth Date: Fri, 4 Oct 2024 17:06:08 +0200 Subject: [PATCH] remove unneeded prints --- totalsegmentator/bin/totalseg_get_modality.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/totalsegmentator/bin/totalseg_get_modality.py b/totalsegmentator/bin/totalseg_get_modality.py index 124a258da..85300e2a8 100644 --- 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