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))