Python createTree 예제들

예제 #1

파일: treeExplore.py 프로젝트: forwho/ml_learning

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()

예제 #2

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()

예제 #3

파일: treeExplore.py 프로젝트: Blockhead3/sevelral-simple-machine-learning-algorithm

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()

예제 #4

#!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)

예제 #5

#!usr/bin/python
#coding=utf8

import regTree
myDat = regTree.loadDataSet('ex00.txt')
retTree = regTree.createTree(myDat)
print retTree