test_train_split ValueError：发现样本数量不一致的输入变量：[200000, 6]答案

【问题标题】：test_train_split ValueError: Found input variables with inconsistent numbers of samples: [200000, 6]test_train_split ValueError：发现样本数量不一致的输入变量：[200000, 6]
【发布时间】：2021-12-10 14:00:42
【问题描述】：

我已经查看了有关此问题的其他几篇帖子，但我无法弄清楚我在这里遇到了什么问题。

我有 X_data 和 Y_data，它们都有形状 (200000,6)。从它们输出的示例数据如下所示：

X_data:
(200000, 6)
[[ 0.00237987  0.00237987 -0.00075756 -0.00221595 -0.00368199  0.00019625]
 [ 0.00171481  0.00171481  0.00176989  0.00125255  0.00275689 -0.00111833]
 [ 0.00190234  0.00190234  0.00333571  0.00127516  0.00146631 -0.00240469]
 ...
 [ 0.00211437  0.00211437  0.00221987  0.0002214   0.00273094 -0.00114419]
 [ 0.00185682  0.00185682  0.00352099  0.00064055 -0.00051575  0.00335213]
 [ 0.00155133  0.00155133 -0.00368774 -0.00200935  0.00225988 -0.00161371]]

Y_data:
(200000, 6)
[[1.         0.14713856 0.04063819 0.03123633 0.00239176 0.01674091]
 [1.         0.35532772 0.09834969 0.19631962 0.0153588  0.10071312]
 [1.         0.17015225 0.04700213 0.04208244 0.00322773 0.02244747]
 ...
 [1.         0.14534398 0.04014234 0.03046259 0.0023313  0.01633189]
 [1.         0.18606737 0.05138638 0.0368341  0.00281708 0.01979553]
 [1.         0.31199003 0.0863072  0.14879644 0.01157114 0.07705023]]

我一做test_train_split，如下：

    ts1 = 0.2
    rs1 = 42
    X_train, X_test, Y_train, Y_test = train_test_split(X_data, Y_data[0], test_size = ts1, random_state = rs1)

我的代码因值错误而崩溃。我不知道我哪里出错了。

【问题讨论】：

请显示错误信息。你的意思是通过Y_data[0] 而不是Y_data？
Y_data[0] 是一个形状为 (6,) 的数组，而 X_data 有 200,000 个样本。
@pavel 和 Frank Yellin 非常感谢。代码的先前迭代需要它是 Y_data[0] 因为 Y_data 具有形状 (1, 200000, 6) 然后。在我发现这个愚蠢的错误之前，我会盯着这个看几个小时。

标签： python pytorch valueerror

【解决方案1】：

您的 Y_data 矩阵的第一列似乎是您的 x 数据的标签，（我不确定您的 Y_train 中的其他 5 列代表什么）。您当前正在获得第一行，这是不正确的（注意大小为 6，但您希望每个 x 输入都有一个 y 值）。所以我认为你想要的代码是

X_train, X_test, Y_train, Y_test = train_test_split(X_data,\
    Y_data[:, 0], test_size = ts1, random_state = rs1)

【讨论】：