diff --git a/aizynthfinder/utils/models.py b/aizynthfinder/utils/models.py index dc8d4ee..1e07f8d 100644 --- a/aizynthfinder/utils/models.py +++ b/aizynthfinder/utils/models.py @@ -105,7 +105,7 @@ def predict(self, *args: np.ndarray, **_: np.ndarray) -> np.ndarray: :param args: the input vectors :return: the vector of the output layer """ - return self.model.predict(args) + return self.model.predict(args, verbose=0) def _log_and_reraise_exceptions(method: Callable) -> Callable: