# apachecn/AiLearning

Switch branches/tags
Nothing to show
Fetching contributors…
Cannot retrieve contributors at this time
125 lines (101 sloc) 3.42 KB
 #!/usr/bin/python # coding:utf8 ''' Created on 2017-03-08 Update on 2017-05-18 Tree-Based Regression Methods Source Code for Machine Learning in Action Ch. 9 Author: Peter/片刻 GitHub: https://github.com/apachecn/AiLearning ''' from __future__ import print_function import regTrees from Tkinter import * from numpy import * import matplotlib from matplotlib.figure import Figure from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg matplotlib.use('TkAgg') def test_widget_text(root): mylabel = Label(root, text="helloworld") # 相当于告诉 布局管理器(Geometry Manager),如果不设定位置，默认在 0行0列的位置 mylabel.grid() # 最大为误差， 最大子叶节点的数量 def reDraw(tolS, tolN): # clear the figure reDraw.f.clf() reDraw.a = reDraw.f.add_subplot(111) # 检查复选框是否选中 if chkBtnVar.get(): if tolN < 2: tolN = 2 myTree = regTrees.createTree(reDraw.rawDat, regTrees.modelLeaf, regTrees.modelErr, (tolS, tolN)) yHat = regTrees.createForeCast(myTree, reDraw.testDat, regTrees.modelTreeEval) else: myTree = regTrees.createTree(reDraw.rawDat, ops=(tolS, tolN)) yHat = regTrees.createForeCast(myTree, reDraw.testDat) # use scatter for data set reDraw.a.scatter(reDraw.rawDat[:, 0].A, reDraw.rawDat[:, 1].A, s=5) # use plot for yHat reDraw.a.plot(reDraw.testDat, yHat, linewidth=2.0, c='red') reDraw.canvas.show() def getInputs(): try: tolN = int(tolNentry.get()) except: tolN = 10 print("enter Integer for tolN") tolNentry.delete(0, END) tolNentry.insert(0, '10') try: tolS = float(tolSentry.get()) except: tolS = 1.0 print("enter Float for tolS") tolSentry.delete(0, END) tolSentry.insert(0, '1.0') return tolN, tolS # 画新的tree def drawNewTree(): # #get values from Entry boxes tolN, tolS = getInputs() reDraw(tolS, tolN) def main(root): # 标题 Label(root, text="Plot Place Holder").grid(row=0, columnspan=3) # 输入栏1, 叶子的数量 Label(root, text="tolN").grid(row=1, column=0) global tolNentry tolNentry = Entry(root) tolNentry.grid(row=1, column=1) tolNentry.insert(0, '10') # 输入栏2, 误差量 Label(root, text="tolS").grid(row=2, column=0) global tolSentry tolSentry = Entry(root) tolSentry.grid(row=2, column=1) # 设置输出值 tolSentry.insert(0,'1.0') # 设置提交的按钮 Button(root, text="确定", command=drawNewTree).grid(row=1, column=2, rowspan=3) # 设置复选按钮 global chkBtnVar chkBtnVar = IntVar() chkBtn = Checkbutton(root, text="Model Tree", variable = chkBtnVar) chkBtn.grid(row=3, column=0, columnspan=2) # 退出按钮 Button(root, text="退出", fg="black", command=quit).grid(row=1, column=2) # 创建一个画板 canvas reDraw.f = Figure(figsize=(5, 4), dpi=100) reDraw.canvas = FigureCanvasTkAgg(reDraw.f, master=root) reDraw.canvas.show() reDraw.canvas.get_tk_widget().grid(row=0, columnspan=3) reDraw.rawDat = mat(regTrees.loadDataSet('db/9.RegTrees/sine.txt')) reDraw.testDat = arange(min(reDraw.rawDat[:, 0]), max(reDraw.rawDat[:, 0]), 0.01) reDraw(1.0, 10) if __name__ == "__main__": # 创建一个事件 root = Tk() # test_widget_text(root) main(root) # 启动事件循环 root.mainloop()