We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent be2894a commit d6e722aCopy full SHA for d6e722a
sklearn/tests/test_qda.py
@@ -3,6 +3,7 @@
3
from sklearn.utils.testing import assert_array_equal
4
from sklearn.utils.testing import assert_array_almost_equal
5
from sklearn.utils.testing import assert_true
6
+from sklearn.utils.testing import assert_greater
7
8
from sklearn import qda
9
@@ -48,9 +49,16 @@ def test_qda():
48
49
50
51
def test_qda_priors():
- clf = qda.QDA(priors=np.array([0.0, 1.0]))
52
+ clf = qda.QDA()
53
+ y_pred = clf.fit(X, y).predict(X)
54
+ n_pos = np.sum(y_pred == 2)
55
+
56
+ neg = 1e-10
57
+ clf = qda.QDA(priors=np.array([neg, 1 - neg]))
58
y_pred = clf.fit(X, y).predict(X)
- assert (y_pred == 2).all()
59
+ n_pos2 = np.sum(y_pred == 2)
60
61
+ assert_greater(n_pos2, n_pos)
62
63
64
def test_qda_store_covariances():
0 commit comments