From df022e35e2a04b6df4ff06881518dad4e5abe0b3 Mon Sep 17 00:00:00 2001 From: nicl-nno Date: Tue, 2 Jul 2024 17:32:52 +0300 Subject: [PATCH] Fix --- fedot/api/time.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/fedot/api/time.py b/fedot/api/time.py index 7d6aa0bc1d..dbaa3a334e 100644 --- a/fedot/api/time.py +++ b/fedot/api/time.py @@ -24,6 +24,7 @@ def __init__(self, **time_params): self.tuning_spend_time = datetime.timedelta(minutes=0) self.assumption_fit_spend_time = datetime.timedelta(minutes=0) + self.assumption_fit_spend_time_single_fold = datetime.timedelta(minutes=0) def __define_timeouts_for_stages(self): """ Determine timeouts for tuning and composing """ @@ -73,7 +74,9 @@ def launch_assumption_fit(self, n_folds: int): """ Wrap assumption fit process with timer """ starting_time_for_assumption_fit = datetime.datetime.now() yield - self.assumption_fit_spend_time = (datetime.datetime.now() - starting_time_for_assumption_fit) * n_folds + self.assumption_fit_spend_time_single_fold = \ + (datetime.datetime.now() - starting_time_for_assumption_fit) * n_folds + self.assumption_fit_spend_time = self.assumption_fit_spend_time_single_fold * n_folds def determine_resources_for_tuning(self): """ @@ -81,7 +84,7 @@ def determine_resources_for_tuning(self): how much time and how many iterations are needed for tuning """ - all_spend_time = self.composing_spend_time + self.assumption_fit_spend_time + all_spend_time = self.composing_spend_time + self.assumption_fit_spend_time_single_fold if self.time_for_automl is not None: all_timeout = float(self.time_for_automl)