为什么在线性回归中使用 train_test_split 方法时出现值错误

Question

在此处输入图片说明

那时我正在执行一些测试我在拆分时遇到值错误

Answer 1

出现此错误是因为您的 X 和 Y 的长度不同（这是 train_test_split 要求的），即 X.shape[0] != Y.shape[0]。 尝试这个：

>>> X.shape

>>> Y.shape

然后解决这个问题：

• 您必须在定义 X 时从 np.array() 内部删除额外的列表，或者之后使用以下命令删除额外的维度：X = X.reshape(X.shape[1:])。
• 然后通过运行 X = X.transpose() 来转置 X 以获得相同数量的 X 和 Y 样本。