def reDraw(tolS, tolN): reDraw.f.clf() reDraw.a = reDraw.f.add_subplot(111) if chkBtnVar.get(): if tolN < 2: tolN = 2 myTree = regTree.createTree(reDraw.rawDat, regTree.modelLeaf, regTree.modelErr, (tolS, tolN)) yHat = regTree.createForeCast(myTree, reDraw.testDat, regTree.modelTreeEval) else: myTree = regTree.createTree(reDraw.rawDat, ops = (tolS, tolN)) yHat = regTree.createForeCast(myTree, reDraw.testDat) reDraw.a.scatter(reDraw.rawDat[ : , 0], reDraw.rawDat[ : , 1], s = 5) reDraw.a.plot(reDraw.testDat, yHat, linewidth = 2.0) reDraw.canvas.show()
def reDraw(tolS, tolN): #画出训练模型的图形 reDraw.f.clf() reDraw.a = reDraw.f.add_subplot(1, 1, 1) if chkBtnVar.get(): if tolN < 2: tolN =2 myTree = regTree.createTree(reDraw.rawDat, regTree.modelLeaf, regTree.modelErr, (tolS, tolN)) yhat = regTree.apiModelTest(myTree, np.mat(reDraw.testDat).T) else: myTree = regTree.createTree(reDraw.rawDat, regTree.regLeaf, regTree.regErr, (tolS, tolN)) yhat = regTree.apiRegTest(myTree, np.mat(reDraw.testDat).T) reDraw.a.scatter(list(map(float, reDraw.rawDat[:, 0])), reDraw.rawDat[:, 1].T.tolist()[0], s = 5) #这里的预测数据原来是mat矩阵,需要转换为列表 reDraw.a.plot(reDraw.testDat, yhat, linewidth = 2.0) reDraw.canvas.show()
def reDraw(tolS,tolN): reDraw.f.clf() reDraw.a=reDraw.f.add_subplot(111) if chkBtnVar.get(): if tolN<2: tolN=2 myTree=regTree.createTree(reDraw.rawDat,regTree.modelLeaf,regTree.modelErr,(tolS,tolN)) yHat=regTree.createForeCast(myTree,reDraw.testDat,regTree.modelTreeEval) else: myTree=regTree.createTree(reDraw.rawDat,ops=(tolS,tolN)) yHat=regTree.createForeCast(myTree,reDraw.testDat) reDraw.a.scatter(reDraw.rawDat[:,0],reDraw.rawDat[:,1],s=5) reDraw.a.plot(reDraw.testDat,yHat,linewidth=2.0) reDraw.canvas.show()
#!usr/bin/env python3 # -*- coding:utf-8 -*- """ #@author:Benny.Chen #@file: main.py #@time: 2020/6/8 21:42 #@email:[email protected] """ import regTree as rt import numpy as np if __name__ == '__main__': myData1 = rt.loadDataSet("ex0.txt") mymat1 = np.mat(myData1) regTree = rt.createTree(mymat1) print(regTree)
#!usr/bin/python #coding=utf8 import regTree myDat = regTree.loadDataSet('ex00.txt') retTree = regTree.createTree(myDat) print retTree