Печать коэффициентов нескольких линейных регрессий на одном скрипте Python
Я бегу этот код:
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]
Есть идеи, почему это так?
1 ответ
Решение
Вам нужно создать отдельные экземпляры вашего классификатора, прежде чем вы вызовете fit:
from sklearn.linear_model import LinearRegression()
model1 = LinearRegression()
model1.fit(bos[IVS],bos.PRICE)
model2 = LinearRegression()
model2.fit(bos[IVS2],bos.PRICE)