当训练集只有一个特征的时候,sklearn的LinearRegression报错

  统计/机器学习 回归分析 Python    浏览次数:3830        分享
0


>>> reg = LinearRegression()
>>> reg.fit(train['x_1'], train['y'])


ValueError: Found input variables with inconsistent numbers of samples

报错内容是inconsistent numbers of samples,但是我检查过,都是482行。

有人知道这是个什么情况嘛?


 

ljljlj   2017-09-15 22:26



   1个回答 
4

我以前遇到过这个。是shape的问题,当输入一个只有一个特征的数据集的时候,sklearn会以为是一个有很多特征的单条数据。

reg.fit(train[['x_1']], train['y'])

或者

reg.fit(train['x_1'].reshape(-1, 1), train['y'])

应该就可以了。


SofaSofa数据科学社区DS面试题库 DS面经

派大星   2017-09-16 00:45

的确可以了,谢谢! - ljljlj   2017-09-19 10:15


  相关主题

提取sklearn逻辑回归的模型系数   1回答

sklearn中除了sgd以外,还有什么可以使用partial_fit方法吗?   1回答

求sklearn.pipeline.Pipeline使用的例子   1回答

sklearn回归的入门级问题   3回答

sklearn predict的默认阈值   2回答

sklearn r2_score返回负数   1回答

sklearn错误ImportError: cannot import name 'joblib'   1回答

查看sklearn版本   1回答

sklearn实现adaboost算法了吗?   1回答

sklearn可以用gpu加速吗?   2回答

sklearn模块有神经网络的类可以调用吗?   1回答

sklearn里怎么做多项式回归?   3回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!