import numpy as np
import pandas as  pd
import pydotplus


from sklearn.datasets import load_iris
iris = load_iris()

from sklearn.cross_validation import train_test_split
# 把數(shù)據(jù)分為測試數(shù)據(jù)和驗證數(shù)據(jù)
train_data, test_data, train_target, test_target = train_test_split(iris.data, iris.target, test_size=0.2,
                                                                    random_state=1)
# Model(建模)-引入決策樹
from sklearn import tree

# 建立一個分類器
clf = tree.DecisionTreeClassifier(criterion="entropy")
# 訓練集進行訓練
clf.fit(train_data, train_target)

# 畫圖方法1-生成dot文件
with open('treeone.dot', 'w') as f:
    dot_data = tree.export_graphviz(clf, out_file=None)
    f.write(dot_data)

# 畫圖方法2-生成pdf文件
dot_data = tree.export_graphviz(clf, out_file=None, feature_names=clf.feature_importances_,
                                filled=True, rounded=True, special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
###保存圖像到pdf文件
graph.write_pdf("treetwo.pdf")

image.png