визуализировать точечные файлы в PyCharm - PullRequest
Новая
       19

визуализировать точечные файлы в PyCharm

0 голосов
/ 16 сентября 2018

Я сгенерировал файл точек для визуализации дерева решений с помощью кода 

import numpy as np
from sklearn.model_selection import train_test_split
import sklearn.tree
from sklearn.datasets import load_breast_cancer

cancer = load_breast_cancer()
X_train, X_test, y_train, y_test =train_test_split(cancer.data,cancer.target, stratify=cancer.target, random_state=42)
tree = sklearn.tree.DecisionTreeClassifier(random_state=0,max_depth=4)
tree.fit(X_train,y_train)
sklearn.tree.export_graphviz(tree,out_file="tree.dot",class_names=cancer.target_names,feature_names=cancer.feature_names,impurity=False, filled=True)

Это успешно создает файл tree.dot. Теперь я могу сгенерировать png-файл с помощью утилиты dot.exe из graphviz (https://graphviz.gitlab.io/_pages/Download/Download_windows.html)

from subprocess import check_call
check_call(['...PATH_TO_GRAPHVIZ/graphviz-2.38/release/bin/dot.exe','-Tpng','tree.dot','-o','tree.png'])

Я хотел бы визуализировать дерево решений также в PyCharm. Есть ли способ сделать это?

...