回归树中的错误讲解math.pow不能对矩阵直接进行操作

最新推荐文章于 2024-05-05 12:15:56 发布

super样儿

最新推荐文章于 2024-05-05 12:15:56 发布

阅读量187

点赞数

分类专栏：机器学习实践文章标签： CART回归树

本文链接：https://blog.csdn.net/weixin_41977181/article/details/89187412

版权

本文讨论了在构建回归树过程中遇到的一个常见错误，即使用math.pow函数时不能直接对矩阵进行操作的问题。

摘要由CSDN通过智能技术生成

第九章回归树@TOC

回归树中的错误讲解math.pow不能对矩阵直接进行操作

import math
import numpy as np
import matplotlib.pyplot as plt

def loadDataSet(fileName):      #general function to parse tab -delimited floats
    dataMat = []                #assume last column is target value
    fr = open(fileName)
    for line in fr.readlines():
        curLine = line.strip().split('\t')
        fltLine = []
        for i in curLine:
           fltLine.append(float(i))
        dataMat.append(fltLine)
    return dataMat

def binSplitDataSet(dataMat,feature,value):
    mat0=dataMat[np.nonzero(dataMat[:,feature]>value)[0]]
    mat1=dataMat[np.nonzero(dataMat[:,feature]<=value)[0]]
    return mat0,mat1

'''
给出建立叶节点的函数
'''
def regLeaf(dataMat):
    return np.mean(dataMat[:,-1])
'''
给出计算误差的函数，总方差=方差*样本总数
'''
def regErr(dataMat):
    return np.var(dataMat[:,-1])*dataMat.shape[0]

'''找到数据集切分的最佳位置
'''
def chooseBestFeature(dataMat,leafType = regLeaf,errType = regErr,ops = (1,4)):
    tolS=ops[0];tolN=ops[1]
    if len(set(dataMat[:,-1].T.tolist()[0])