《李宏毅机器学习》task9

正在熬夜编写中。。。

使用Matplotlib注释
构造注释树

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', va='center', ha='center', bbox=nodeType, arrowprops=arrow_args)

def createPlot():
     fig = plt.figure(1, facecolor='white') #创建新的figure 1, 背景颜色为白色
     fig.clf()  #清空figure 1的内容    '''
     createPlot.ax1 = plt.subplot(111, frameon=False)
     plotNode('a decision node', (0.5, 0.1), (0.1, 0.5),  decisionNode)
     plotNode('a leaf node', (0.8, 0.1), (0.3, 0.8), leafNode)
     plt.show()

if __name__ =="__main__":
    createPlot()