多元线性回归与scipy linregress

iklwldmw  于 5个月前  发布在  其他
关注(0)|答案(1)|浏览(73)

我正在尝试训练一个非常简单的线性回归模型。
我的代码是:

from scipy import stats

xs = [[   0,    1,  153]
 [   1,    2,    0]
 [   2,    3,  125]
 [   3,    1,   93]
 [   2,   24, 5851]
 [   3,    1,  524]
 [   4,    1,    0]
 [   2,    3,    0]
 [   2,    1,    0]
 [   5,    1,    0]]

ys = [1, 1, 1, 1, 1, 0, 1, 1, 0, 1]

slope, intercept, r_value, p_value, std_err = stats.linregress(xs, ys)

字符串
我得到以下错误:

File "/System/Library/Frameworks/Python.framework/Versions/2.7/Extras/lib/python/scipy/stats/stats.py", line 3100, in linregress
ssxm, ssxym, ssyxm, ssym = np.cov(x, y, bias=1).flat
File "/System/Library/Frameworks/Python.framework/Versions/2.7/Extras/lib/python/numpy/lib/function_base.py", line 1747, in cov
X = concatenate((X, y), axis)
ValueError: all the input array dimensions except for the concatenation 
axis must match exactly


我的输入有什么问题?我尝试了几种方法来改变ys的结构,但都不起作用。

olqngx59

olqngx591#

您正在寻找多元回归。AFAIK stats.linregress没有此功能。
您可能想尝试sklearn.linear_model.LinearRegression。检查this答案。

相关问题