預測新客戶是否會流失
就像你在影片中看到的,若要使用 sklearn 訓練模型:
- 匯入你要使用的模型——這裡是 Support Vector Classifier:
from sklearn.svm import SVC
- 進行實例化:
svc = SVC()
- 將模型訓練(也就是「fit」)到資料上:
svc.fit(telco['data'], telco['target'])
這裡第 1 個引數是特徵,第 2 個引數是我們想要預測的標籤——也就是客戶是否會流失。擬合模型後,你可以使用模型的 .predict() 方法來預測新客戶的標籤。
不論你用哪一種模型,流程都一樣,而 sklearn 提供了很多模型!在這個練習中,你會使用 LogisticRegression。
本練習屬於課程