bad input shape (125, 2),拆分训练和测试
2019-08-10 本文已影响0人
三盏灯亮一盏
#导入数据集生成工具
from sklearn.datasets import make_blobs
#导入手机拆分工具
from sklearn.model_selection import train_test_split
#生成样本数量为500,分类数为5的数据集
X,y=make_blobs(n_samples=500,centers=5,random_state=8)
#将数据集拆分为训练集和测试集
X_train,y_train,X_test,y_test=train_test_split(X,y,random_state=8)
#使用贝努力贝叶斯拟合数据
nb=BernoulliNB()
nb.fit(X_train,y_train)
#打印模型得分
print('模型得分:{:.3f}'.format(nb.score(X_test,y_test)))
ValueError: bad input shape (125, 2)
正确顺序是:X_train,X_test,y_train,y_test