当前位置:   article > 正文

解决ValueError: bad input shape (xxxx, x)_python bad input shape

python bad input shape

下面看一段机器学习代码:

  1. from sklearn import datasets
  2. from sklearn.model_selection import train_test_split
  3. from sklearn.naive_bayes import GaussianNB
  4. from sklearn.model_selection import cross_val_score
  5. iris=datasets.load_iris()
  6. x_train,y_train,x_test,y_test = train_test_split(iris.data,iris.target,test_size=0.4,random_state=0)
  7. clf=GaussianNB()
  8. clf.fit(x_train,y_train)

使用高斯朴素贝叶斯模型对鸢尾花数据集进行训练,但运行时出现以下问题:

查看错误原因,提示为值错误,这个错误发生在模型训练位置,并且模型训练语句没有错误,那么就是传给模型训练的数据发生了错误(x_train,y_train)!!!我们返回去检查这两个数据集,发现使用train_test_split在分割数据集时,我们将数据集顺序写错了,正确的数据集顺序为:x_train,x_test,y_train,y_test,正是由于这个正确的顺序,才能分割正确的数据集与测试集,大部分人认为这个数据集没有顺序,下面看一下train_test_split的常规用法:

X_train,X_test, y_train, y_test =sklearn.model_selection.train_test_split(train_data,train_target,test_size=0.4, random_state=0)

# train_data:所要划分的样本特征集

# train_target:所要划分的样本结果

# test_size:样本占比,如果是整数的话就是样本的数量

# random_state:是随机数的种子。

因此千万不要忽略训练集与测试集的顺序!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/316147
推荐阅读
相关标签
  

闽ICP备14008679号