O'Reilly logo

Einführung in Machine Learning mit Python by Andreas C. Müller, Sarah Guido

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

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 ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required