决策树可视化

from sklearn import tree

with open('iris01.dot','w') as writer:
    tree.export_graphviz(decision_tree=alog,out_file=writer)

其中alog为：

alog=DecisionTreeClassifier()
alog.fit(x_train,y_train)

alog即为训练好的训练好的决策树模型。

然后在pycharm中进入terminal终端，进入到刚才保存的dot文件的目录，然后将保存的dot格式转换为pdf格式，在终端中输入如下命令：

dot -Tpdf iris01.dot -o iris01.pdf