신규 고객의 이탈 여부 예측하기

영상에서 보신 것처럼, sklearn로 모델을 학습하려면 다음과 같이 진행해요:

• 사용할 모델을 임포트합니다. 여기서는 Support Vector Classifier를 사용해요:

from sklearn.svm import SVC

• 인스턴스를 생성합니다:

svc = SVC()

• 데이터를 사용해 모델을 학습(fit)합니다:

svc.fit(telco['data'], telco['target'])

여기서 첫 번째 인자는 특성(feature)이고, 두 번째 인자는 우리가 예측하려는 레이블(label)입니다. 즉, 고객이 이탈할지 여부죠. 모델을 학습한 후에는 모델의 .predict() 메서드를 사용해 신규 고객의 레이블을 예측할 수 있어요.

이 과정은 어떤 모델을 사용하든 동일하며, sklearn에는 사용할 수 있는 모델이 아주 많습니다! 이번 연습에서는 LogisticRegression을 사용해 보겠습니다.