Skip to content
Snippets Groups Projects
Commit c86fc38d authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Disable progress bar printing from similarity forest regressor training

parent 29a11860
No related branches found
No related tags found
1 merge request!12Resolve "integration-sota"
This commit is part of merge request !12. Comments created here will be created in the context of that merge request.
......@@ -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]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment