From c86fc38dcad4acdc390cebc46db99918ea6c179a Mon Sep 17 00:00:00 2001
From: Charly Lamothe <charly.lamothe@univ-amu.fr>
Date: Fri, 6 Mar 2020 05:49:09 +0100
Subject: [PATCH] Disable progress bar printing from similarity forest
regressor training
---
code/bolsonaro/models/similarity_forest_regressor.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/code/bolsonaro/models/similarity_forest_regressor.py b/code/bolsonaro/models/similarity_forest_regressor.py
index 8f02daf..647e869 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]
--
GitLab