fix: add missing property to logistic regression class

iamDecode · Jul 9, 2021 · 4b6e11c · 4b6e11c
1 parent ef98c12
commit 4b6e11c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)