import numpy as np
from sklearn.naive_bayes import MultinomialNB
#导入数据集生成工具
from sklearn.datasets import make_blobs
#导入数据集拆分工具
from sklearn.model_selection import train_test_split
X,y=make_blobs(n_samples=500,centers=5,random_state=8)
X_train,X_test,y_train,y_test=train_test_split(X,y,random_state=8)
#导入数据预处理工具
from sklearn.preprocessing import MinMaxScaler
scaler=MinMaxScaler()
scaler.fit(X_train)
X_train_scaled=scaler.transform(X_train)
X_test_scaled=scaler.transform(X_test)
mnb=MultinomialNB()
mnb.fit(X_train_scaled,y_train)
print('模型得分:{:.3f}'.format(mnb.score(X_test_scaled, y_test)))
#导入画图工具
import matplotlib.pyplot as plt
#限定坐标轴范围,X共两列
x_min,x_max=X[:,0].min()-0.5,X[:,0].max()+0.5
y_min,y_max=X[:,1].min()-0.5,X[:,1].max()+0.5
#使用不同的背景色表示不同的分类
xx,yy = np.meshgrid(np.arange(x_min, x_max,.02),np.arange(y_min, y_max, .02))
z = mnb.predict(np.c_[(xx.ravel(),yy.ravel())]).reshape(xx.shap
多项式朴素贝叶斯【机器学习算法一朴素贝叶斯4】
最新推荐文章于 2024-05-23 15:43:21 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)