Einführung in Machine Learning mit Python

Analysieren von Entscheidungsbäumen

Bäume lassen sich über die Funktion export_graphviz aus dem Modul tree visualisieren. Diese schreibt eine Datei im Format .dot, einem Textformat zum Speichern von Graphen. Wir setzen einen Parameter zum Einfärben der Knoten, um die mehrheitliche Kategorie in jedem Knoten zu kennzeichnen, und übergeben die Namen von Kategorien und Merkmalen, sodass der Baum dementsprechend beschriftet werden kann:

In[58]:

from sklearn.tree import export_graphviz

export_graphviz(tree, out_file="tree.dot", class_names=["malignant", "benign"],

feature_names=cancer.feature_names, impurity=False, filled=True)

Wir können diese Datei lesen und mit dem Modul graphviz wie in Abbildung 2-27 gezeigt visualisieren (Sie können aber auch ...