From de8b672cc6c9fe715c7a2bd10ac3b5c3b4bd2b7f Mon Sep 17 00:00:00 2001 From: Lopa10ko Date: Fri, 17 Jan 2025 18:31:16 +0300 Subject: [PATCH] fix: add compute_config layer on output_folder and history dir setting --- benchmark/abstract_bench.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/benchmark/abstract_bench.py b/benchmark/abstract_bench.py index 43379d0d2..b8ecedaa4 100644 --- a/benchmark/abstract_bench.py +++ b/benchmark/abstract_bench.py @@ -57,10 +57,8 @@ def run(self): def evaluate_loop(self, dataset, experiment_setup: dict = None): matplotlib.use('TkAgg') train_data, test_data = DataLoader(dataset_name=dataset).load_data() - experiment_setup['output_folder'] = experiment_setup['output_folder'] + \ - f'/{dataset}' - experiment_setup['history_dir'] = './composition_results' + \ - f'/{dataset}' + experiment_setup['compute_config']['output_folder'] += f'/{dataset}' + experiment_setup['compute_config']['history_dir'] = f'./composition_results/{dataset}' model = FedotIndustrial(**experiment_setup) model.fit(train_data) prediction = model.predict(test_data)