Anzeigen von RandomForestRegression-Bäumen

Question

Ich habe ein RandomForestRegression-Modell in sklearn ausgeführt und die Ausgabe meiner Entscheidungsbäume (n_estimators = 50) in 50 .dot-Dateien gespeichert.

Jetzt möchte ich sie speichern, damit ich sie als tatsächliche Bäume anzeigen kann.

Ich versuche dies:

import pydotplus 

dot_data=r'F:\Sheyenne\Random_Forest\my_tree0.dot' 

graph = pydotplus.graph_from_dot_data(dot_data) 

graph.write_pdf(r'F:\Sheyenne\Random_Forest\my_tree0.pdf') 

aber zurück:

AttributeError: 'NoneType' object has no attribute 'write_pdf' 

Answer 1

Sieht aus wie Sie eine Datei zu laden versuchen. Versuchen Sie folgendes:

import pydotplus 

dot_file=r'F:\Sheyenne\Random_Forest\my_tree0.dot' 

graph = pydotplus.graph_from_dot_file(dot_file) 

graph.write_pdf(r'F:\Sheyenne\Random_Forest\my_tree0.pdf')