美文网首页
bad input shape (125, 2),拆分训练和测试

bad input shape (125, 2),拆分训练和测试

作者: 三盏灯亮一盏 | 来源:发表于2019-08-10 11: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

相关文章

网友评论

      本文标题:bad input shape (125, 2),拆分训练和测试

      本文链接:https://www.haomeiwen.com/subject/aznljctx.html