def reDraw(tolS,tolN):
reDraw.f.clf() # clear the figure
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)
reDraw.a.scatter(reDraw.rawDat[:,0].tolist(), reDraw.rawDat[:,1].tolist(), s=5) #use scatter for data set
reDraw.a.plot(reDraw.testDat, yHat, linewidth=2.0) #use plot for yHat
reDraw.canvas.show()
def reDraw(tolS,tolN):
reDraw.f.clf() # clear the figure
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)
reDraw.a.scatter(reDraw.rawDat[:,0], reDraw.rawDat[:,1], s=5) # use scatter for data set
reDraw.a.plot(reDraw.testDat, yHat, linewidth=2.0) # use plot for yHat
reDraw.canvas.show()
