实现贝叶斯的代码

import numpy as np
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report,confusion_matrix
from sklearn.naive_bayes import GaussianNB,BaseDiscreteNB,MultinomialNB
#加载数据
iris=datasets.load_iris()
x_train,x_text,y_train,y_text=train_test_split(iris.data,iris.target)#分片
#print(x_train)
mul_nb=GaussianNB()#这里可以替换 BaseDiscreteNB或者MultinomialNB
mul_nb.fit(x_train,y_train)
print(classification_report(mul_nb.predict(x_text),y_text))
print(confusion_matrix(mul_nb.predict(x_text),y_text))

猜你喜欢

转载自blog.csdn.net/zhuiyunzhugang/article/details/105883993