statsmodels 회귀분석

파이썬에서 회귀분석은 statsmodels 라이브러리를 사용

* OLS(Ordinary Least Squares)
* 잔차제곱합(RSS : Residual Sum of Squares)을 최소화하는 방식

 

 

1. from_formula( ) 함수를 이용하여 정의한 공식(formula) 을 활용해 회귀분석 정의 가능

formula = """
MEDV ~ scale(CRIM) + scale(ZN) + scale(INDUS) + scale(NOX) + scale(RM) + scale(AGE) + scale(DIS) + scale(RAD) + scale(TAX) + scale(PTRATIO) + scale(B) + scale(LSTAT)
"""

 

2. sm.OLS.from_formula() 메서드를 사용하여 선형 회귀 모델을 생성

import statsmodels.api as sm

model = sm.OLS.from_formula(formula, data=train)
result = model.fit()