手动实现交叉验证遇DataConversionWarning报错，如何解决？

嗨，这个报错我之前手动实现交叉验证时也碰到过，其实本质就是sklearn的模型对目标变量y的形状有明确要求——它期望y是一维数组（形状为(n_samples,)），但你传进去的是列向量（形状为(n_samples, 1)）。给你几个简单靠谱的解决办法：

• 使用ravel()方法（报错提示里已经提到的方案）：直接将列向量转为一维数组，代码示例：

  import numpy as np
  # 假设y是你的目标变量列向量
  y = y.ravel()
  

• 使用flatten()方法：和ravel()效果类似，区别是flatten()会返回数组的副本，而ravel()优先返回视图（更节省内存），代码示例：

  y = y.flatten()
  

• 通过索引提取一维数据：如果y是numpy数组或pandas对象，可直接通过索引取第一列转为一维：

  # numpy数组的情况
  y = y[:, 0]
  # pandas DataFrame的情况
  y = y.iloc[:, 0].values
  

• 从源头避免列向量：如果是从pandas DataFrame中提取目标变量，直接用values获取一维数组：

  # 假设df是你的数据集，target是目标列名
  y = df['target'].values
  

处理完后你可以用print(y.shape)检查一下，只要形状变成(n_samples,)，这个警告就会消失啦。

内容的提问来源于stack exchange，提问作者Nc123