Den Baum exportieren

Bei Klassifikationsaufgaben mit Decision Trees ist Overfitting meist die Folge sehr tief gewachsener Bäume. Der Vergleich der Accuracy-Werte auf Trainings- und Testdaten zeigt, dass du Overfitting in deinen Ergebnissen hast. Das lässt sich auch an der Baumvisualisierung erkennen.

In dieser Übung exportierst du den Decision Tree in ein Textdokument, das anschließend zur Visualisierung verwendet werden kann.

Diese Übung ist Teil des Kurses

<Kurs>HR Analytics: Mitarbeiterfluktuation mit Python vorhersagen</Kurs>

Kurs ansehen

Übungsanweisungen

Importiere die Funktion export_graphviz() aus dem Untermodul sklearn.tree.
Trainiere das model mit den Trainingsdaten.
Exportiere die Visualisierung in die Datei tree.dot.

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# Import the graphical visualization export function
from sklearn.____ import export_graphviz

# Apply Decision Tree model to fit Features to the Target
model.____(features_train,target_train)

# Export the tree to a dot file
____(model,"tree.____")

Code bearbeiten und ausführen