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

ćwiczenie

Analiza reguł drzewa decyzyjnego

W tym ćwiczeniu wyodrębnisz reguły warunkowe z drzewa decyzyjnego i zwizualizujesz je, aby zidentyfikować główne czynniki wpływające na odejście klientów.

Wytrenowana instancja drzewa decyzyjnego jest załadowana jako mytree, a wyskalowane cechy – jako pandas DataFrame o nazwie train_X. Moduł tree z biblioteki sklearn oraz biblioteka graphviz zostały już dla ciebie załadowane.

Uwaga: zamiast display(graph) użyto własnej funkcji display_image(), aby ułatwić ci podgląd wyników.

Instrukcje

100 XP
  • Wyeksportuj obiekt graphviz z wytrenowanego drzewa decyzyjnego.
  • Przypisz nazwy cech.
  • Ustaw precyzję na 1 i dodaj nazwy klas.
  • Wywołaj funkcję Source() z biblioteki graphviz i przekaż jej wyeksportowany obiekt graphviz.