Печать коэффициентов нескольких линейных регрессий на одном скрипте 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)
Другие вопросы по тегам