From 7269af500c7ff37c1f9d626f58a85cc0a93f90e7 Mon Sep 17 00:00:00 2001 From: Lopa10ko Date: Tue, 23 Jul 2024 16:46:10 +0300 Subject: [PATCH] feat: add target param for tuning function --- fedot/api/main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/fedot/api/main.py b/fedot/api/main.py index eb00faa9fa..f389489acc 100644 --- a/fedot/api/main.py +++ b/fedot/api/main.py @@ -203,6 +203,7 @@ def fit(self, def tune(self, input_data: Optional[FeaturesType] = None, + target: TargetType = 'target', metric_name: Optional[Union[str, MetricCallable]] = None, iterations: int = DEFAULT_TUNING_ITERATIONS_NUMBER, timeout: Optional[float] = None, @@ -213,6 +214,7 @@ def tune(self, Args: input_data: data for tuning pipeline in one of the supported formats. + target: data target values in one of the supported target formats. metric_name: name of metric for quality tuning. iterations: numbers of tuning iterations. timeout: time for tuning (in minutes). If ``None`` or ``-1`` means tuning until max iteration reach. @@ -230,7 +232,7 @@ def tune(self, if input_data is None: input_data = self.train_data else: - input_data = self.data_processor.define_data(features=input_data, target=self.target, is_predict=False) + input_data = self.data_processor.define_data(features=input_data, target=target, is_predict=False) cv_folds = cv_folds or self.params.get('cv_folds') n_jobs = n_jobs or self.params.n_jobs