⬆️ version 0.5.5 and 🔧 fix logistic regression parameters

mljar · Jul 22, 2020 · 247045c · 247045c
1 parent 13a9cff
commit 247045c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name='mljar-supervised',
-    version='0.5.4',
+    version='0.5.5',
     description='Automated Machine Learning for Supervised tasks',
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/supervised/__init__.py b/supervised/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "0.5.4"
+__version__ = "0.5.5"
 
 from supervised.automl import AutoML
diff --git a/supervised/algorithms/linear.py b/supervised/algorithms/linear.py
@@ -30,7 +30,7 @@ def __init__(self, params):
         logger.debug("LinearAlgorithm.__init__")
         self.max_iters = 1
         self.library_version = sklearn.__version__
-        self.model = LogisticRegression(max_iter=1000, n_jobs=-1)
+        self.model = LogisticRegression(max_iter=500, tol=5e-4, n_jobs=-1)
 
     def file_extension(self):
         return "linear"