Prédire si un nouveau client se désabonnera

Comme vous l'avez vu dans la vidéo, pour entraîner un modèle avec sklearn :

• Importez le modèle qui vous intéresse — ici, un classifieur à vecteurs de support :

from sklearn.svm import SVC

• Instanciez-le :

svc = SVC()

• Entraînez-le, ou « fittez-le », sur les données :

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

Ici, le premier argument correspond aux variables explicatives (features), et le deuxième est l'étiquette que l'on cherche à prédire — si le client se désabonnera ou non. Une fois le modèle ajusté, vous pouvez utiliser la méthode .predict() du modèle pour prédire l'étiquette d'un nouveau client.

Ce processus est le même peu importe le modèle utilisé, et sklearn en offre beaucoup ! Dans cet exercice, vous utiliserez LogisticRegression.