Merge pull request #32 from Alex-Lekov/develop

v1.3.7
Alex-Lekov · Mar 5, 2021 · a73a170 · a73a170
2 parents d47570d + f56fa18
commit a73a170
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git 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
@@ -1 +1 @@
-__version__ = "1.3.6"
+__version__ = "1.3.7"
diff --git 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
@@ -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,
         )