在单个 python 脚本上打印多个线性回归的系数

Question

我是运行这个代码：

import sklearn
IVS=['CRIM', 'RM', 'PTRATIO']
IVS2=['CRIM','PTRATIO']
model1=lm.fit(bos[IVS],bos.PRICE)
model2=lm.fit(bos[IVS2],bos.PRICE)

print(model1.coef_)
print(model2.coef_)

当尝试打印出两个模型的系数时，我只得到了两个打印函数的最后一个模型系数：

[-0.27939868 -1.83737204]
[-0.27939868 -1.83737204]

知道这是为什么吗？

Answer 1

您需要在调用 fit 之前创建单独的分类器实例：

from sklearn.linear_model import LinearRegression()
model1 = LinearRegression()
model1.fit(bos[IVS],bos.PRICE)
model2 = LinearRegression()
model2.fit(bos[IVS2],bos.PRICE)