test_naive_bayes.py 文件源码

def test_basic(self, single_chunk_binary_classification):
        X, y = single_chunk_binary_classification
        a = nb.PartialBernoulliNB(classes=[0, 1])
        b = nb_.BernoulliNB()
        a.fit(X, y)
        b.partial_fit(X, y, classes=[0, 1])
        assert_eq(a.coef_, b.coef_)