sklearn naive_bayes

239 阅读1分钟

适用条件

1特征独立
2特征重要性相等
from sklearn import datasets
iris = datasets.load_iris()
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()
clf = clf.fit(iris.data, iris.target)
y_pred=clf.predict(iris.data)
import numpy as np
np.mean(y_pred == iris.target)
0.96
from sklearn.naive_bayes import MultinomialNB
clf = MultinomialNB()
clf = clf.fit(iris.data, iris.target)
y_pred=clf.predict(iris.data)
np.mean(y_pred == iris.target)
0.9533333333333334
from sklearn.naive_bayes import BernoulliNB
clf = BernoulliNB()
clf = clf.fit(iris.data, iris.target)
y_pred=clf.predict(iris.data)
np.mean(y_pred == iris.target)
0.3333333333333333