sklearn中的朴素贝叶斯模型及其应用

高斯分布型

from sklearn import datasets
iris= datasets.load_iris()

from sklearn.naive_bayes import GaussianNB
gnb = GaussianNB()
pred = gnb.fit(iris.data,iris.target)
y_pred = pred.predict(iris.data)

print(iris.data.shape[0],(iris.target!=y_pred).sum())

  

多项式型

from sklearn import datasets
iris= datasets.load_iris()

from sklearn.naive_bayes import MultinomialNB
gnb = MultinomialNB()
pred = gnb.fit(iris.data,iris.target)
y_pred = pred.predict(iris.data)

print(iris.data.shape[0],(iris.target!=y_pred).sum())

  

 

伯努利型

from sklearn import datasets
iris= datasets.load_iris()

from sklearn.naive_bayes import BernoulliNB
gnb = BernoulliNB()
pred = gnb.fit(iris.data,iris.target)
y_pred = pred.predict(iris.data)

print(iris.data.shape[0],(iris.target!=y_pred).sum())

  

 

 2.使用sklearn.model_selection.cross_val_score(),对模型进行验证。

from sklearn.naive_bayes import GaussianNB
from sklearn.model_selection import cross_val_score
gnb = GaussianNB()

scores = cross_val_score(gnb,iris.data,iris.target,cv=10)
print("Accuracy:%.3f"%scores.mean())

  

 

posted on 2018-11-26 09:41  zoyeln  阅读(232)  评论(0编辑  收藏  举报

导航