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

上一篇 下一篇

猜你喜欢

热点阅读