ValueError:找到输入样本数量不一致的输入变量:[559,140]

时间:2019-03-04 05:20:59

标签: machine-learning classification

这是下面的代码...我不知道我的代码有什么问题。请帮助。错误发生在行中 clf.fit(X_train,y_train)

Details

1 个答案:

答案 0 :(得分:2)

问题出在您的 X_train,y_train,X_test,y_test = train_test_split(X,y,test_size = 0.2)部分。

根据here中的scikit-learn文档, train_test_split 函数的正确返回值顺序为:

  • X_train,
  • X_test,
  • y_train,
  • y_test

您在代码中的顺序是错误的。让我们用以下行替换您使用** train_test_split **的行:

X_train, X_test, y_train, y_test = train_test_split(X,y, test_size=0.2)

希望这能解决您的问题。