From 90e0b11426a7632dfbd9658e145b9483350d6258 Mon Sep 17 00:00:00 2001 From: Ravin Kohli Date: Mon, 22 Nov 2021 14:22:57 +0100 Subject: [PATCH 1/2] set verbose=False in catboost --- .../setup/traditional_ml/traditional_learner/learners.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py b/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py index eeb1094f1..7f4077936 100644 --- a/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py +++ b/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py @@ -135,7 +135,11 @@ def _fit(self, X_train: np.ndarray, X_train_pooled = Pool(data=X_train, label=y_train, cat_features=categoricals) X_val_pooled = Pool(data=X_val, label=y_val, cat_features=categoricals) - self.model.fit(X_train_pooled, eval_set=X_val_pooled, use_best_model=True, early_stopping_rounds=early_stopping) + self.model.fit(X_train_pooled, + eval_set=X_val_pooled, + use_best_model=True, + early_stopping_rounds=early_stopping, + verbose=False) @staticmethod def get_properties( From ec8235f24f22160968ea39960c485d3a86c79655 Mon Sep 17 00:00:00 2001 From: Ravin Kohli Date: Mon, 22 Nov 2021 14:25:04 +0100 Subject: [PATCH 2/2] fix flake --- .../setup/traditional_ml/traditional_learner/learners.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py b/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py index 7f4077936..220c52dcd 100644 --- a/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py +++ b/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py @@ -136,10 +136,10 @@ def _fit(self, X_train: np.ndarray, X_val_pooled = Pool(data=X_val, label=y_val, cat_features=categoricals) self.model.fit(X_train_pooled, - eval_set=X_val_pooled, - use_best_model=True, - early_stopping_rounds=early_stopping, - verbose=False) + eval_set=X_val_pooled, + use_best_model=True, + early_stopping_rounds=early_stopping, + verbose=False) @staticmethod def get_properties(