diff --git a/CHANGELOG.md b/CHANGELOG.md index 90480aa..1d0e1d0 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file. The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/). +## [1.3.7] +### Fix +- target encoder in model.opt + + ## [1.3.6] ### ADD - add dosc on CV diff --git a/automl_alex/__version__.py b/automl_alex/__version__.py index 9c9ce67..f0ffdf3 100644 --- a/automl_alex/__version__.py +++ b/automl_alex/__version__.py @@ -1 +1 @@ -__version__ = "1.3.6" \ No newline at end of file +__version__ = "1.3.7" \ No newline at end of file diff --git a/automl_alex/_base.py b/automl_alex/_base.py index 9a35d9f..05517ac 100644 --- a/automl_alex/_base.py +++ b/automl_alex/_base.py @@ -269,9 +269,8 @@ def opt( self.optimizer = BestSingleModel( type_of_estimator=self._type_of_estimator, - models_names=[ - self.__name__, - ], + models_names=[self.__name__], + target_encoders_names=[], feature_selection=False, auto_parameters=auto_parameters, folds=folds, diff --git a/automl_alex/automl_alex.py b/automl_alex/automl_alex.py index e84a616..c9a3b44 100644 --- a/automl_alex/automl_alex.py +++ b/automl_alex/automl_alex.py @@ -485,7 +485,8 @@ def fit( time.sleep(0.1) self.model_5 = automl_alex.BestSingleModel( - models_names=["LightGBM", "ExtraTrees"], + models_names=["LightGBM", "ExtraTrees", "RandomForest"], + target_encoders_names=["CatBoostEncoder",], feature_selection=feature_selection, **params, )