From 4b6e11cf5f96151eb9fa49e58c3a89ee1c870ca2 Mon Sep 17 00:00:00 2001 From: Dennis Collaris Date: Fri, 9 Jul 2021 13:15:37 +0200 Subject: [PATCH] fix: add missing property to logistic regression class --- sklearn_pmml_model/linear_model/implementations.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sklearn_pmml_model/linear_model/implementations.py b/sklearn_pmml_model/linear_model/implementations.py index 70074c6..9109899 100644 --- a/sklearn_pmml_model/linear_model/implementations.py +++ b/sklearn_pmml_model/linear_model/implementations.py @@ -108,6 +108,7 @@ def __init__(self, pmml): self.coef_ = np.array(self.coef_) self.intercept_ = np.array(self.intercept_) self.multi_class = 'auto' + self.solver = 'lbfgs' def fit(self, x, y): return PMMLLinearClassifier.fit(self, x, y)