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)