决策树图形展示

最新推荐文章于 2023-01-01 16:48:01 发布

一名路过的小码农啊

最新推荐文章于 2023-01-01 16:48:01 发布

阅读量2.7k

点赞数 1

分类专栏：理论知识

本文链接：https://blog.csdn.net/lovely_girl1126/article/details/79169322

版权

绘图工具，基于python的Matplotlib，绘出决策树图形，以下代码亲测可用

从原始数据集中创建决策树，并使用python函数库绘制树形图。

#coding=UTF-8
import matplotlib.pyplot as plt
decisionNode = dict(boxstyle="sawtooth",fc="0.8")
leafNode=dict(boxstyle="round4",fc="0.8")
arrow_args=dict(arrowstyle="<-")

def plotNode(nodeTxt,centerPt,parentPt,nodeType):
	createPlot.ax1.annotate(nodeTxt,xy=parentPt,\
		xycoords='axes fraction',
		xytext=centerPt,textcoords='axes fraction',\
		va="center",ha="center",bbox=nodeType,arrowprops=arrow_args)

def createPlot():
	fig=plt.figure(1,facecolor='white')
	fig.clf()
	createPlot.ax1=plt.subplot(111,frameon=False)
	plotNode('decisive node',(0.5,0.1),(0.1,0.5),decisionNode)
	plotNode('leaf node ',(0.8,0.1),(0.3,0.8),leafNode)
	plt.show()
	
def get