Skip to content

Commit 8eae3b5

Browse files
committed
Increase sklearn model max_iter to 10000 to allow convergence in new versions of scikit-learn
1 parent e5938a5 commit 8eae3b5

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

tests/integration/test_pymas.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,11 @@ def sklearn_model(train_data):
5858
X, y = train_data
5959
with warnings.catch_warnings():
6060
warnings.simplefilter("ignore")
61-
model = LogisticRegression(multi_class="multinomial", solver="lbfgs")
61+
model = LogisticRegression(
62+
multi_class="multinomial",
63+
solver="lbfgs",
64+
max_iter=10000
65+
)
6266
model.fit(X, y)
6367
return model
6468

tests/integration/test_tasks.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,11 @@ def sklearn_logistic_model():
3939

4040
with warnings.catch_warnings():
4141
warnings.simplefilter("ignore")
42-
model = LogisticRegression(multi_class="multinomial", solver="lbfgs")
42+
model = LogisticRegression(
43+
multi_class="multinomial",
44+
solver="lbfgs",
45+
max_iter=10000
46+
)
4347
model.fit(iris.iloc[:, 0:4], iris["Species"])
4448

4549
return model, iris.iloc[:, 0:4]

0 commit comments

Comments
 (0)