Merge pull request #82 from compomics/feature/verbosity

Pass verbosity to `predict` and DeepLC Retrainer
compomics · Sep 19, 2024 · 8e7c4f8 · 8e7c4f8
2 parents 3740720 + 149d9d7
commit 8e7c4f8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py
@@ -610,7 +610,7 @@ def make_preds_core(
         try:
             X
             ret_preds = mod.predict(
-                [X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf
+                [X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf, verbose=int(self.verbose)
             ).flatten()
         except UnboundLocalError:
             logger.debug("X is empty, skipping...")
@@ -1170,6 +1170,7 @@ def calibrate_preds(
                 freeze_layers=True,
                 n_epochs=self.n_epochs,
                 freeze_after_concat=1,
+                verbose=self.verbose,
             )
 
             self.model = models