Hotfix for bug

aimclub · Jul 30, 2024 · f684be4 · f684be4
1 parent 81acea3
commit f684be4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fedot/preprocessing/preprocessing.py b/fedot/preprocessing/preprocessing.py
@@ -207,6 +207,7 @@ def _prepare_obligatory_unimodal(self, data: InputData, source_name: str,
         if is_fit_stage:
             self._find_features_lacking_nans(data, source_name)
         self._take_only_correct_features(data, source_name)
+
         if is_fit_stage:
             data = self._drop_rows_with_nan_in_target(data)
 
@@ -220,6 +221,7 @@ def _prepare_obligatory_unimodal(self, data: InputData, source_name: str,
             data.target = self._apply_target_encoding(data, source_name)
         else:
             self.types_correctors[source_name].convert_data_for_predict(data)
+            data.target = self._apply_target_encoding(data, source_name)
 
         # TODO andreygetmanov target encoding must be obligatory for all data types
         if data_type_is_text(data):