LinearRegression训练数据集维度有限制吗？

来源：5-9 使用scikit-learn解决回归问题

qq_山人

2021-09-12

sklearn中的标准线性回归训练方法中，训练集不能使用一维数组么？使用05小节中的数据（取boston.data[:, 5]作为训练数据），会提示报错Expected 2D array, got 1D array instead

写回答

2回答

liuyubobobo

2021-09-13

不可以使用一维数据，必须是二维数据。即使只有一个特征，也要是 m * 1 的二维数据。

X_train.reshape(-1, 1)也不行

请给我相对完整能展现出问题的代码片段或者截图。比如 X_train 是什么？什么叫不行，报的错是什么。

一个简单的验证方式是看一下 X_train.shape。

继续加油！：）

qq_山人

原因找到了，是predict时没有对X_test也进行reshape操作导致的错误。应该在最开始split数据前对原始数据提前reshape，就不需要后面再多次reshape了

2021-09-13

共1条回复

qq_山人

提问者

2021-09-12

X_train.reshape(-1, 1)也不行

Python3入门机器学习经典算法与应用

Python3+sklearn，兼顾原理、算法底层实现和框架使用。

5940 学习 · 2455 问题

相似问题

回答 1

回答 2

回答 1

回答 1

回答 1