Skip to content

Commit

Permalink
fix progress bar
Browse files Browse the repository at this point in the history
  • Loading branch information
MorrisNein committed Nov 13, 2023
1 parent 3b3233c commit cb77e9c
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions experiments/fedot_warm_start/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -375,8 +375,9 @@ def main():
# Gathering knowledge base
train_histories = {}
with open(progress_file_path, 'a') as progress_file:
for dataset_id in (pbar := tqdm(dataset_ids, 'FEDOT, all datasets', file=progress_file)):
pbar.set_description(pbar.desc + f' ({dataset_id})')
description = 'FEDOT, all datasets'
for dataset_id in (pbar := tqdm(dataset_ids, description, file=progress_file)):
pbar.set_description(description + f' ({dataset_id})')
try:
timeout = TRAIN_TIMEOUT if dataset_id in dataset_ids_test else TEST_TIMEOUT
dataset = algorithm.components.datasets_loader.load_single(dataset_id)
Expand All @@ -397,8 +398,9 @@ def main():
pickle.dump(algorithm, meta_learner_file)

with open(progress_file_path, 'a') as progress_file:
description = 'MetaFEDOT, Test datasets'
for dataset_id in (pbar := tqdm(dataset_ids_test, 'MetaFEDOT, Test datasets', file=progress_file)):
pbar.set_description(pbar.desc + f' ({dataset_id})')
pbar.set_description(description + f' ({dataset_id})')
try:
# Run meta AutoML
# 1
Expand Down

0 comments on commit cb77e9c

Please sign in to comment.