diff --git a/code/bolsonaro/models/similarity_forest_regressor.py b/code/bolsonaro/models/similarity_forest_regressor.py index 8f02daf0200e55a9021a44ab70cdef8a6fd2f935..647e8695da88c0f84817a602471fd90f9bd1f1b0 100644 --- a/code/bolsonaro/models/similarity_forest_regressor.py +++ b/code/bolsonaro/models/similarity_forest_regressor.py @@ -38,12 +38,12 @@ class SimilarityForestRegressor(BaseEstimator, metaclass=ABCMeta): val_scores.append(tree.predict(X_val)) tree_pred_bar.update(1) - with tqdm(range(self._extracted_forest_size), disable=False) as pruning_forest_bar: + with tqdm(range(self._extracted_forest_size), disable=True) as pruning_forest_bar: pruning_forest_bar.set_description(f'[Pruning forest s={self._extracted_forest_size}]') for i in pruning_forest_bar: best_similarity = 100000 found_index = 0 - with tqdm(range(len(tree_list)), disable=False) as tree_list_bar: + with tqdm(range(len(tree_list)), disable=True) as tree_list_bar: tree_list_bar.set_description(f'[Tree selection s={self._extracted_forest_size} #{i}]') for j in tree_list_bar: lonely_tree = tree_list[j]