Визуализация деревьев RandomForestRegression

Question

Я запустил модель RandomForestRegression в sklearn и сохранил выходные данные деревьев решений (n_estimators = 50) в 50 .dot файлах.

Теперь я хочу сохранить их, чтобы я мог рассматривать их как фактические деревья.

Я пытаюсь это:

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') 

но это возвращает:

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

Answer 1

Похоже, вы пытаетесь загрузить файл. Попробуйте следующее:

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')