python - Sklearn train_test_split 创建一维数组

Question

密切关注 scikit-learn 文档，我正在尝试安装一个虚拟分类器。但是，运行时会引发值错误。这是出乎意料的，因为之前我使用相同的数据:X = vector_data(稀疏矩阵)，y = vector_target(int列表)与网格搜索cv拟合，没有这样的错误。

所以肯定是train_test_split的引入。

为什么会在我的代码中发生这种情况，而不是在文档中，请问应该采取哪些预防措施？

X_train, X_test, y_train, y_test = train_test_split(vector_data, vector_target, random_state=0)
clf = DummyClassifier(strategy='stratified',random_state=0)
clf.fit(X_train, y_train)
clf.score(X_test,y_test)

 DeprecationWarning: Passing 1d arrays as data is deprecated in 0.17 and will raise ValueError in 0.19.
Reshape your data either using X.reshape(-1, 1) if your data has a single feature or X.reshape(1, -1) if it contains a single sample.
      DeprecationWarning)

Answer 1

首先，始终，始终，阅读警告/错误消息。

所以，尝试这样的事情:

X_train, X_test, y_train, y_test = train_test_split(vector_data.reshape(-1, 1), 
             vector_target, random_state=0)

我不能说更多，只是尝试理解这段代码。