1. Nauka
    2. /
  2. Kursy
    3. /
  3. Uczenie maszynowe w marketingu w Pythonie
Connected

ćwiczenie

Dopasowanie modelu drzewa decyzyjnego

Teraz dopasujemy drzewo decyzyjne na zbiorze treningowym zbioru danych telekomunikacyjnych, a następnie przewidzimy etykiety na niewidzianych wcześniej danych testowych i obliczymy dokładność prognoz modelu. Zobaczysz, jak różni się jego wydajność w porównaniu z regresją logistyczną.

Funkcja accuracy_score została już zaimportowana, a zbiory treningowy i testowy zbudowane wcześniej są dostępne jako train_X i test_X dla cech oraz train_Y i test_Y dla zmiennych docelowych.

Instrukcje

100 XP
  • Zainicjalizuj klasyfikator drzewa decyzyjnego.
  • Dopasuj drzewo decyzyjne na danych treningowych.
  • Przewidź etykiety odejścia klientów na danych testowych.
  • Oblicz i wyświetl wynik dokładności na danych testowych.