我已经生成了一个点文件来使用代码可视化决策树
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 文件。我现在可以使用 graphviz ( https://graphviz.gitlab.io/_pages/Download/Download_windows.html ) 的 dot.exe 实用程序生成一个 png 文件
from subprocess import check_call
check_call(['...PATH_TO_GRAPHVIZ/graphviz-2.38/release/bin/dot.exe','-Tpng','tree.dot','-o','tree.png'])
我也想在 PyCharm 中可视化决策树。有没有办法做到这一点？
最佳答案
你可以安装bzixilu的dotplugin插件，
当您打开点文件时，图表将自动显示在它旁边
关于pycharm - 在 PyCharm 中可视化点文件，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/52352836/