diff --git a/setup.py b/setup.py index 867472a4..8a9a7bec 100644 --- 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 index c1804bd8..7eb26d10 100644 --- 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 index 9d2831e9..1f1a9688 100644 --- 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"