1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Marketing Analytics: Predicting Customer Churn in Python
Connected

cvičení

Trénování dalšího modelu ze scikit-learn

Všechny modely z sklearn mají metody .fit() a .predict(), stejně jako model LogisticRegression, který jsi použil/a v předchozím cvičení. Díky tomu můžeš snadno vyzkoušet různé modely a zjistit, který podává nejlepší výsledky. Aby sis lépe osvojil/a práci s sklearn API, v tomto cvičení zkusíš místo LogisticRegression natrénovat DecisionTreeClassifier.

Pokyny

100 XP
  • Importuj DecisionTreeClassifier z sklearn.tree.
  • Vytvoř instanci klasifikátoru a výsledek ulož do proměnné clf.
  • Natrénuj klasifikátor na datech. Příznaky jsou uloženy v proměnné features a cílová proměnná je 'Churn'.
  • Předpověz label pro new_customer.