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

HR Analytics: Mitarbeiterfluktuation mit Python vorhersagen

Kurs anzeigen

Anleitung zur Übung

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 Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# 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