2017-03-01 5 views
0

Ich habe einen Arbeitsentscheidungsbaum und auch eine funktionierende .Dot-Ausgabedatei aus meinen Trainingsdaten. Ich möchte den Entscheidungsbaumprozess für jeden Datensatz ausgeben. Was ich meine ist, dass ich dem Benutzer eher als einen Baum mit einem einzelnen Zweig oder sogar als Text zeigen möchte, welchen Weg jeder Datensatz verfolgt hat, um zu dem entschiedenen Ergebnis zu kommen. Ist das möglich?Scikit Learn - Entscheidungsbaum - visuelle Darstellung des Ergebnisses jedes Datensatzes

Antwort

1

Sie können die decision_path Methode (here) des trainierten Baums verwenden, um den Pfad zu sehen. Sie können jedoch nur die apply Methode (here) benötigen. Die apply Methode zeigt Ihnen den Blattknoten. Da es nur einen Weg gibt, um zu jedem Blatt zu gelangen, kann man den Pfad erkennen, wenn man das Blatt kennt und den Baum sieht.

Wie Sie dies anzeigen möchten, liegt bei Ihnen und dem Benutzer. Eine Tabelle, eine Visualisierung usw.

+0

Das ist großartig. Ich möchte potenziell zwei Methoden anzeigen, einmal wäre eine Visualisierung, die andere wäre eine Tabelle. Für einen Tisch wäre es Leaf | Value – HMan06