TST Use random state to initialize MLPClassifier. (#12892)

scikit-learn · Jan 27, 2019 · 3207278e0313449bc42372ab0312fa250fd63bc8 · 3207278
1 parent 1f5bcae
commit 3207278e0313449bc42372ab0312fa250fd63bc8
Unified Split

Showing with 2 additions and 1 deletion.

+2 −1 sklearn/neural_network/tests/test_mlp.py
diff --git a/sklearn/neural_network/tests/test_mlp.py b/sklearn/neural_network/tests/test_mlp.py
@@ -432,7 +432,8 @@ def test_predict_proba_binary():
     X = X_digits_binary[:50]
     y = y_digits_binary[:50]
 
-    clf = MLPClassifier(hidden_layer_sizes=5)
+    clf = MLPClassifier(hidden_layer_sizes=5, activation='logistic',
+                        random_state=1)
     with ignore_warnings(category=ConvergenceWarning):
         clf.fit(X, y)
     y_proba = clf.predict_proba(X)