Skip to content

Commit a95c1b2

Browse files
martinoywajnothman
authored andcommitted
EXA Fixed Convergence Warnings by changing solver to 'lbfgs' in plot_mlp_alpha.py (scikit-learn#14158)
1 parent fded644 commit a95c1b2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/neural_networks/plot_mlp_alpha.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@
3636

3737
classifiers = []
3838
for i in alphas:
39-
classifiers.append(MLPClassifier(alpha=i, random_state=1))
39+
classifiers.append(MLPClassifier(solver='lbfgs', alpha=i, random_state=1,
40+
hidden_layer_sizes=[100, 100]))
4041

4142
X, y = make_classification(n_features=2, n_redundant=0, n_informative=2,
4243
random_state=0, n_clusters_per_class=1)

0 commit comments

Comments
 (0)