linearRegression（）返回列表中的列表（sklearn）

Question

我在Python中进行多元线性回归，但由于某种原因，系数没有正确返回为列表。相反，将返回列表中的列表：

from sklearn import linear_model
clf = linear_model.LinearRegression()
# clf.fit ([[0, 0, 0], [1, 1, 1], [2, 2, 2]], [0, 1, 2])
clf.fit([[394, 3878, 13, 4, 0, 0],[384, 10175, 14, 4, 0, 0]],[3,9])
print 'coef array',clf.coef_
print 'length', len(clf.coef_)
print 'getting value 0:', clf.coef_[0]
print 'getting value 1:', clf.coef_[1]

这将返回列表[]的列表中的值，而不是列表[]。知道为什么会这样吗？输出：

coef array [[  1.03428648e-03   9.54477167e-04   1.45135995e-07   0.00000000e+00
0.00000000e+00   0.00000000e+00]]
length 1
getting value 0: [  1.03428648e-03   9.54477167e-04   1.45135995e-07   0.0000000
0e+00 0.00000000e+00   0.00000000e+00]
getting value 1:
Traceback (most recent call last):
  File "regress.py", line 8, in <module>
    print 'getting value 1:', clf.coef_[1]
IndexError: index out of bounds

但这行得通：

from sklearn import linear_model
clf = linear_model.LinearRegression()
clf.fit ([[0, 0, 0], [1, 1, 1], [2, 2, 2]], [0, 1, 2])
# clf.fit([[394, 3878, 13, 4, 0, 0],[384, 10175, 14, 4, 0, 0]],[3,9])
print 'coef array',clf.coef_
print 'length', len(clf.coef_)
print 'getting value 0:', clf.coef_[0]
print 'getting value 1:', clf.coef_[1]

输出：

coef array [ 0.33333333  0.33333333  0.33333333]
length 3
getting value 0: 0.333333333333
getting value 1: 0.333333333333

Answer 1

似乎是scipy.linalg.的一个问题如果你跟踪调用链，它首先在https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/linear_model/base.py#L218，然后在https://github.com/scipy/scipy/blob/master/scipy/linalg/basic.py#L468.到达if语句。在第一种情况下m，n=2,6，在第二种情况下，您有m，n=3,3。

Answer 2

我从未使用过你提到的多元线性回归模型，所以我不知道为什么会发生这种情况。但是，如果您只是想解决您的问题，您可以将列表展平：

flat_list = clf.coef_[0]

<罢工> 如果列表可能有多个子列表（并且您想将它们组合成一个扁平列表），那么您可以使用更通用的方法来扁平化它：

flat_list = [item for sublist in clf.coef_ for item in sublist]

编辑：在等待软件包开发人员的真实解释/解决方案时，您可以依赖如下解决方案：

if isinstance(clf.coef_[0], list):
    clf.coef_ = clf.coef_[0]

只有当列表中有子列表时，才会将其展平。

Answer 3

这是通过更新SciKit学习文件夹中的两个文件修复的。

代码如下：https://github.com/scikit-learn/scikit-learn/commit/d0b20f0a21ba42b85375b1fbc7202dc3962ae54f