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

Question

在此处输入图像描述

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

标签： machine-learninglinear-regressionpredictionvalueerror

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 中相同数量的样本。