test_naive_bayes.py 文件源码

def test_gaussiannb():
    iris = load_iris()
    clf = GaussianNB()
    clf.fit(iris.data, iris.target)
    y_pred = clf.predict(iris.data)
    print(y_pred)
    clf_ = SKGaussianNB()
    clf_.fit(iris.data, iris.target)
    print(clf_.predict(iris.data))


    print(iris.target)