python决策树 math库CART算法
每周一搏,提升自我。 本篇博文主要是对math的CART算法总结,参照博文:http://www.cnblogs.com/wsine/p/5180321.html 代码如下: #!/usr/bin/python #coding:utf-8 ''' CART决策树算法 ''' import matplotlib.pyplot as plt #决策树属性设置 decisionNode=dict(boxstyle="sawtooth",fc="0.8") leafNode=dict(boxstyle="round4",fc="0.8") arrow_args=dict(arrowstyle="<-") #createPlot 主函数,调用即可画出决策树,其中调用登了剩下的所有的函数,inTree的形式必须为嵌套的决策树 def createPlot(inThree): fig=plt.figure(1,facecolor='white') fig.clf() axprops=dict(xticks=[],yticks=[]) createPlot.ax1=plt.subplot(111,frameon=False,**axprops) #no ticks # createPlot.ax1=plt.subplot(111,frameon=False) #ticks for demo