使用pydotplus&GraphViz绘制决策树图形

最新推荐文章于 2024-04-23 19:23:54 发布

weixin_33834075

最新推荐文章于 2024-04-23 19:23:54 发布

阅读量1.1k

点赞数

文章标签：数据结构与算法 python 人工智能

原文链接：https://my.oschina.net/kyo4321/blog/3009740

版权

2019独角兽企业重金招聘Python工程师标准>>>

###安装GraphViz之后，需要设置环境变量，还需配套安装pydotplus
https://blog.csdn.net/chai_zheng/article/details/78226556


###葡萄酒公开数据集下载地址：http://archive.ics.uci.edu/ml/datasets/Wine

###随机森林可视化
https://blog.csdn.net/linhai1028/article/details/79827331


import time
import pydotplus
import numpy as np
from sklearn import tree
from sklearn.externals.six import StringIO
from sklearn.model_selection import train_test_split

print('Step 1.Loading data...')
data = np.loadtxt("D:/data/Wine.txt", delimiter=',')
x = data[:, 1:14]
y = data[:, 0].reshape(178,1)
X_train, X_test, Y_train, Y_test = train_test_split(x, y, test_size=0.4)
print('---Loading and splitting completed.')


print('Step 2.Training...')
startTime = time.time()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X_train,Y_train)
print('---Training Completed.Took %f s.'%(time.time()-startTime))


print('Step 3.Testing...')
Y_predict = clf.predict(X_test)
matchCount = 0
for i in range(len(Y_predict)):
    if Y_predict[i] == Y_test[i]:
        matchCount += 1
accuracy = float(matchCount/len(Y_predict))
print('---Testing completed.Accuracy: %.3f%%'%(accuracy*100))


feature_name = ['Alcohol','Malic Acid','Ash','Alcalinity of Ash','Magnesium','Total Phenols',
                'Flavanoids','Nonflavanoid Phenols','Proantocyanins','Color Intensity','Hue',
                'OD280/OD315 of Diluted Wines','Proline']
target_name = ['Class1','Class2','Class3']


dot_data = StringIO()

tree.export_graphviz(clf,out_file = dot_data,feature_names=feature_name,
                     class_names=target_name,filled=True,rounded=True,
                     special_characters=True)


graph = pydotplus.graph_from_dot_data(dot_data.getvalue())

graph.write_pdf("WineTree.pdf")
print('Visible tree plot saved as pdf.')

转载于:https://my.oschina.net/kyo4321/blog/3009740