Scikit-learn의 LinearRegression() 함수

LinearRegression() 

scikit-learn 패키지를 사용하여 선형 회귀분석을 하는 경우 linear_model 서브 패키지의 LinearRegression 클래스를 사용한다.

 

• LinearRegression 클래스 객체 생성

from sklearn.linear_model import LinearRegression

lrmodel = LinearRegression()
lrmodel.fit(X_train, Y_train)

fit 메서드로 모형 추정. 상수항 결합을 자동으로 해주므로 사용자가 직접 add_constant 등의 명령를 써서 상수항 결합을 할 필요는 없다.

 

• coef

추정된 가중치 벡터

가중치들을 보여준다.

lrmodel.coef_

 

• intercept

추정된 상수항

절편을 나타낸다.

lrmodel.intercept_

 

•  _ 는 머신러닝에서 유도된 결과를 나타내는 기호이다.

 

 

 

 

 

---

참고

데이터 사이언스 스쿨_선형 회귀분석의 기초

scikit-learn_LinearRegression