더북(TheBook)

7.4.6 로지스틱 회귀 모델의 설정 및 훈련

앞서 임포트한 Scikit-learn 라이브러리의 로지스틱 함수 LogisticRegression을 실행해 모델을 만든다. 그리고 fit 함수를 이용해 훈련용 예측 데이터와 목표 데이터를 적합시킨다. 즉, 이는 모델을 만드는 과정이다.

model = LogisticRegression( )
model = model.fit( X_train, y_train )

# 회귀계수는 model.coef_에 저장된다. 다음과 같이 계수를 확인한다
pd.DataFrame( zip( X.columns, np.transpose( model.coef_ ) ) )
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.