1. Nauka
    2. /
  2. Kursy
    3. /
  3. HR Analytics: Przewidywanie rotacji pracowników w Pythonie
Connected

ćwiczenie

Eksportowanie drzewa

W zadaniach klasyfikacji za pomocą drzew decyzyjnych nadmierne dopasowanie (overfitting) jest zazwyczaj wynikiem zbyt głęboko rozbudowanych drzew. Porównanie wyników dokładności na zbiorze treningowym i testowym pokazuje, że w twoich wynikach właśnie to zjawisko występuje. Można się o tym przekonać również na podstawie wizualizacji drzewa.

W tym ćwiczeniu wyeksportujesz drzewo decyzyjne do pliku tekstowego, który następnie posłuży do wizualizacji.

Instrukcje

100 XP
  • Zaimportuj funkcję export_graphviz() z submodułu sklearn.tree.
  • Dopasuj model do danych treningowych.
  • Wyeksportuj wizualizację do pliku tree.dot.