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