File tree Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -58,7 +58,11 @@ def sklearn_model(train_data):
58
58
X , y = train_data
59
59
with warnings .catch_warnings ():
60
60
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
+ )
62
66
model .fit (X , y )
63
67
return model
64
68
Original file line number Diff line number Diff line change @@ -39,7 +39,11 @@ def sklearn_logistic_model():
39
39
40
40
with warnings .catch_warnings ():
41
41
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
+ )
43
47
model .fit (iris .iloc [:, 0 :4 ], iris ["Species" ])
44
48
45
49
return model , iris .iloc [:, 0 :4 ]
You can’t perform that action at this time.
0 commit comments