采用多项式分布
from sklearn.naive_bayes import MultinomialNB multinomial = MultinomialNB() multinomial.fit(train,target) y_ = multinomial.predict(X_test) plt.scatter(X_test[:,0],X_test[:,1],c=y_) cmap = ListedColormap(['r','g','b']) plt.scatter(train[:,0],train[:,1],c=target,cmap=cmap)