sklean中决策树的可视化

from sklearn.datasets import load_iris
from sklearn import tree
import graphviz
import sys
import pydotplus


def visualize(clf,path):
	dot_data = tree.export_graphviz(clf, out_file=None,feature_names=get_feature_names("./data/csv/E17.csv"))
    #dot_data = tree.export_graphviz(clf, out_file=None)
    graph = pydotplus.graph_from_dot_data(dot_data)
    graph.write_pdf(path+"iris.pdf")


iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
visualize(clf,"./output/")

with open("./output/iris.dot", 'w') as f:
    tree.export_graphviz(clf, out_file=f)