fix test error

rasbt · Nov 25, 2019 · c5c6ae4 · c5c6ae4
1 parent 1eed6aa
commit c5c6ae4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mlxtend/classifier/tests/test_stacking_classifier.py b/mlxtend/classifier/tests/test_stacking_classifier.py
@@ -522,7 +522,9 @@ def test_decision_function():
     np.random.seed(123)
 
     # PassiveAggressiveClassifier has no predict_proba
-    meta = PassiveAggressiveClassifier(random_state=42)
+    meta = PassiveAggressiveClassifier(max_iter=1000,
+                                       tol=0.001,
+                                       random_state=42)
     clf1 = RandomForestClassifier(n_estimators=10)
     clf2 = GaussianNB()
     sclf = StackingClassifier(classifiers=[clf1, clf2],
@@ -537,4 +539,4 @@ def test_decision_function():
                              cv=5,
                              scoring='roc_auc')
     scores_mean = (round(scores.mean(), 2))
-    assert scores_mean == 0.96, scores_mean
+    assert scores_mean == 0.93, scores_mean