已有环境pycharm
1.安装graphviz
在终端输入:apt-get install graphviz,可能有的同学提示权限不足,则输入 sudo apt-get install graphviz后按回车,输入计算机密码,此时输入密码是没有任何显示的。
2.可以直接使用pycharm安装依赖包graphviz,另一种方法在终端输入pip install graphviz进行安装。
完成以上两个步骤之后可以跑一个demo代码测试:
#-*- coding: utf-8 -*-
from sklearn import datasets
from sklearn.ensemble import RandomForestClassifier
from sklearn import tree
from IPython.display import Image
import pydotplus
# 仍然使用自带的iris数据
iris = datasets.load_iris()
X = iris.data
y = iris.target
# 训练模型,限制树的最大深度4
clf = RandomForestClassifier(max_depth=4)
#拟合模型
clf.fit(X, y)
Estimators = clf.estimators_
for index, model in enumerate(Estimators):
filename = 'iris_' + str(index) + '.pdf'
dot_data = tree.export_graphviz(model , out_file=None,
feature_names=iris.feature_names,
class_names=iris.target_names,
filled=True, rounded=True,
special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
Image(graph.create_png())
graph.write_pdf(filename)
可能有些同学运行该代码还会出现依赖包不存在的情况,则根据各自不同的情况安装没有的依赖包即可。
运行后的实验结果如图所示,在图片的左栏会生成iris_0.pdf等10个文件:
打开iris_0.pdf,结果如图所示:
如有问题欢迎留言。