Skip to content

Commit 8e7c4f8

Browse files
Merge pull request #82 from compomics/feature/verbosity
Pass verbosity to `predict` and DeepLC Retrainer
2 parents 3740720 + 149d9d7 commit 8e7c4f8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

deeplc/deeplc.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -610,7 +610,7 @@ def make_preds_core(
610610
try:
611611
X
612612
ret_preds = mod.predict(
613-
[X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf
613+
[X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf, verbose=int(self.verbose)
614614
).flatten()
615615
except UnboundLocalError:
616616
logger.debug("X is empty, skipping...")
@@ -1170,6 +1170,7 @@ def calibrate_preds(
11701170
freeze_layers=True,
11711171
n_epochs=self.n_epochs,
11721172
freeze_after_concat=1,
1173+
verbose=self.verbose,
11731174
)
11741175

11751176
self.model = models

0 commit comments

Comments
 (0)