Predicția abandonului pentru un client nou

Așa cum ai văzut în video, pentru a antrena un model cu sklearn:

• Importă modelul dorit – în acest caz, un Support Vector Classifier:

from sklearn.svm import SVC

• Instanțiază-l:

svc = SVC()

• Antrenează-l, sau „potrivește-l" pe date:

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

Aici, primul argument conține caracteristicile, iar al doilea argument reprezintă eticheta pe care încercăm să o prezicem – dacă un client va abandona sau nu. După ce ai potrivit modelul, poți folosi metoda .predict() pentru a prezice eticheta unui client nou.

Acest proces este același indiferent de modelul ales, iar sklearn oferă foarte multe! În acest exercițiu, vei folosi LogisticRegression.