diff --git a/asreviewcontrib/models/dynamic_nn.py b/asreviewcontrib/models/dynamic_nn.py index b7e7055..060f682 100644 --- a/asreviewcontrib/models/dynamic_nn.py +++ b/asreviewcontrib/models/dynamic_nn.py @@ -40,7 +40,8 @@ def fit(self, X, y): if X.shape[1] > max_features: raise ValueError(f"Feature size too large: {X.shape[1]} features. Maximum allowed is {max_features}.") - print("\nNumber of features:", X.shape[1]) + if self.verbose == 1: + print("\nNumber of features:", X.shape[1]) # Determine the number of layers based on the number of rows num_layers = ceil(log10(max(10, X.shape[0]))) @@ -53,7 +54,8 @@ def fit(self, X, y): min_delta=self.min_delta, restore_best_weights=True) - print(f"\nFitting New Iteration with {num_layers} layers:\n") + if self.verbose == 1: + print(f"\nFitting New Iteration with {num_layers} layers:\n") self._model.fit(X, y, epochs=100, batch_size=32, shuffle=True, callbacks=[callback])