CART:分类与回归树

起源:决策树切分数据集

决策树每次决策时,按照一定规则切分数据集,并将切分后的小数据集递归处理。这样的处理方式给了线性回归处理非线性数据一个启发。

能不能先将类似特征的数据切成一小部分,再将这一小部分放大处理,使用线性的方法增加准确率呢?

 

Part I:  树的枝与叶

枝:二叉 or 多叉?

   在AdaBoost的单决策树中,对于连续型数据构建决策树,我们采取步进阈值切分2段的方法。还有一种简化处理,即选择子数据集中的当前维度所有不同的值作为阈值切分。

而在CART里,大于阈值归为左孩子,小于阈值的归为右孩子。若是离散型数据,则根据离散数据种类建立对应的多叉树即可。

叶:何时不再切分?

ID3决策树中,停止切分的条件有两个:

      ①DFS链路中全部切分方式被扫过一次,很明显,对于离散型特征,每次按照某一维度异同切分,再扫相同维度毫无意义。

          对于连续型特征,则共有维度*(当前维度不同值数量,即阈值数量)种切分方式,同种方式也毫无意义。

      ②当前子数据集分类全部一致,已经是很完美的切分了,再切也没意思。

CART中,由于搜索深度只有1,重复选取也不会卡死。所以直接遵循②。

追加③条件:手动限制切分子集数量下限tolN,误差变化下限tolS(目标函数收敛)。一旦达到这两个下限,就立刻停止。

枝:一个好枝?

ID3算法给出了一个评价离散Label的好枝的标准:分类的混乱度(香农熵)降低。

对于连续数据,好枝的参考标准则是类似最小二乘法的目标函数,即误差越小越好。

由于计算误差需要先进行线性回归,相当于树套回归,虽然效果很好,但是无疑带来计算压力。

在这点上, CART利用均值和方差的性质给出了一个简化的误差计算:即假设一团数据的回归结果是这团数据的均值,那么目标函数即可当成总方差。

使用均值替代回归结果的树称为回归树,使用实际回归结果的树成为模型树。

叶:数量越多越好?

叶结点数量越多,越容易过拟合。数量越少,则容易欠拟合。

而tolN和tolS在选择最好的切分方式时,控制着叶结点的数量,这两个值越小,叶子越多,且对tolS的值很敏感。

树的递归构建:

①对当前数据集做最好的切分。

②若不能切分,则将该结点设为叶结点。

否则,由于切分的性质,所以切出的两个子集必定不为空。对大于阈值的子集进行左孩子递归构建,小于阈值的子集进行右孩子递归构建。

 

Part II :  树的剪枝

叶结点数量决定着拟合情况。人工调整不是一件好事。

所以出现一种先强行过拟合(tolN=0,tolS=1)生成CART树,然后利用新的样本数据进行剪枝的方法,称为后剪枝。

后剪枝有两种方法:

①后剪枝会将大量的枝从树顶直接转化成叶子,相当于废掉原树中很多数据,所以需要引入新的数据。

而把一个大枝转为叶子的方法,则是利用均值的性质。新叶子的回归值=原枝上所有叶的均值。

②除了废枝为叶,还有利用均值的计算性质、借助新数据归并两叶。当然归并是有条件的。

新数据递归切分之后,必然会分到叶子上。如果恰好一个枝上是两片叶子,那么分别计算ErrNoMerga、ErrMerga的值,观察是否变小来决定是否归并。

$ErrNoMerga=\sum_{i=1}^{LSet}(Set[i].y-L.leaf)^{2}+\sum_{i=1}^{RSet}(Set[i].y-R.leaf)^{2}$

$NewLeaf=mergaMean=avg(L.leaf+R.leaf)$

$ErrMerga=\sum_{i=1}^{Set}{(Set[i].y-mergaMean)^{2}}$

 

Part III:   回归与模型树

对于每条测试数据,从树顶按照树中保存的切分规则左右递归直到叶结点,返回叶结点的值作为回归值。

实际测试结果下,效果并不好。所以应当每一个叶结点:保留数据、以及线性回归方程(w、b),从而建立起模型树。

线性模型树方法将取代回归树中的均值误差理论,主要修改地方在选择分支、后剪枝上。

$Err =\sum_{i=1}^{m} (data[i].y-Regression(y))^{2}$

这样,叶结点就变成了一个线性回归器,返回线性回归结果即可。

 

Part IV 代码

#include "cstdio"

#include "iostream"

#include "fstream"

#include "math.h"

#include "sstream"

#include "string"

#include "vector"

#include "set"

using namespace std;

#define Dim dataSet[0].feature.size()

#define TREE pair<vector<Data>,vector<Data> >

#define NULL 0

struct Data

{

    vector<double> feature;

    double y;

    Data(vector<double> feature,double y):feature(feature),y(y) {}

};

struct RegTree

{

    int dim;double value;

    RegTree *Left,*Right;

    RegTree():Left(NULL),Right(NULL) {}

    RegTree(int dim,double value):Left(NULL),Right(NULL),dim(dim),value(value) {}

};

vector<Data> dataSet,addSet,testSet;

pair<int,double> ops(0,1);

void read()

{

    ifstream fin("data1.txt"),fin2("data2.txt"),fin3("data3.txt");

    string line;double tmp,y;

    while(getline(fin,line))

    {

        stringstream sin(line);

        vector<double> feature;

        while(sin>>tmp) feature.push_back(tmp);

        y=feature.back();feature.pop_back();

        dataSet.push_back(Data(feature,y));

    }

    while(getline(fin2,line))

    {

        stringstream sin(line);

        vector<double> feature;

        while(sin>>tmp) feature.push_back(tmp);

        y=feature.back();feature.pop_back();

        addSet.push_back(Data(feature,y));

    }

    while(getline(fin3,line))

    {

        stringstream sin(line);

        vector<double> feature;

        while(sin>>tmp) feature.push_back(tmp);

        y=feature.back();feature.pop_back();

        testSet.push_back(Data(feature,y));

    }

}

pair<vector<Data>,vector<Data> > splitDataSet(vector<Data> dataSet,int dim,double value)

{

    vector<Data> Left,Right;

    for(int i=0;i<dataSet.size();i++)

    {

        if(dataSet[i].feature[dim]>value) Left.push_back(dataSet[i]);

        else Right.push_back(dataSet[i]);

    }

    return make_pair(Left,Right);

}

double regLeaf(vector<Data> dataSet)

{

    double ret=0.0;

    //printf("Leaf:\n");

    for(int i=0;i<dataSet.size();i++)

    {

        ret+=dataSet[i].y;

        /*

        for(int j=0;j<dataSet[i].feature.size();j++) printf("%.2lf ",dataSet[i].feature[j]);

        printf("%lf\n",dataSet[i].y);*/

    }

    //printf("\n");

    return ret/dataSet.size();

}

double calcErr(vector<Data> dataSet)

{

    double avg=0.0,ret=0.0;

    for(int i=0;i<dataSet.size();i++) avg+=dataSet[i].y;

    avg/=dataSet.size();

    for(int i=0;i<dataSet.size();i++) ret+=(dataSet[i].y-avg)*(dataSet[i].y-avg);

    return ret;

}

pair<int,double> chooseBestSplit(vector<Data> dataSet)

{

    //tolN、tolS(较敏感)过小都会导致Leaf过多,过大则会导致Leaf过少

    int tolN=ops.first;double tolS=ops.second,S,newS,bestS=1e10,bestValue,bestDim;

    set<double> y;

    for(int i=0;i<dataSet.size();i++) y.insert(dataSet[i].y);

    if(y.size()==1) return make_pair(-1,regLeaf(dataSet));

    S=calcErr(dataSet);

    for(int i=0;i<Dim;i++)

    {

        set<double> splitValue;

        for(int j=0;j<dataSet.size();j++) splitValue.insert(dataSet[j].feature[i]);

        for(set<double>::iterator j=splitValue.begin();j!=splitValue.end();j++)

        {

            TREE tree=splitDataSet(dataSet,i,*j);

            if(tree.first.size()<tolN||tree.second.size()<tolN) continue;

            newS=calcErr(tree.first)+calcErr(tree.second);

            if(newS<bestS) {bestDim=i;bestValue=*j;bestS=newS;}

        }

    }

    if(S-bestS<tolS) return make_pair(-1,regLeaf(dataSet));

    TREE tree=splitDataSet(dataSet,bestDim,bestValue);

    if(tree.first.size()<tolN||tree.second.size()<tolN) return make_pair(-1,regLeaf(dataSet));

    return make_pair(bestDim,bestValue);

}

RegTree *buildTree(vector<Data> dataSet)

{

    pair<int,double> info=chooseBestSplit(dataSet);

    if(info.first==-1)

    {

        RegTree *node=new RegTree(info.first,info.second);

        return node;

    }

    RegTree *node=new RegTree(info.first,info.second);

    TREE tree=splitDataSet(dataSet,info.first,info.second);

    //printf("Node: dim:%d %.2lf\n",info.first,info.second);

    node->Left=buildTree(tree.first);

    node->Right=buildTree(tree.second);

    return node;

}

double getMean(RegTree *root)

{

    double ret=0.0;

    if(root->Left->dim!=-1) ret+=getMean(root->Left);

    else ret+=root->Left->value;

    if(root->Right->dim!=-1) ret+=getMean(root->Right);

    else ret+=root->Right->value;

    return ret/=2;

}

RegTree *prune(RegTree *&root,vector<Data> dataSet)

{

    if(dataSet.size()==0) return new RegTree(-1,getMean(root));

    double errNoMerga=0.0,errMerga=0.0;

    if(root->Left->dim!=-1||root->Right->dim!=-1)

    {

        TREE tree=splitDataSet(dataSet,root->dim,root->value);

        if(root->Left->dim!=-1) root->Left=prune(root->Left,tree.first);

        if(root->Right->dim!=-1) root->Right=prune(root->Right,tree.second);

    }

    if(root->Left->dim==-1&&root->Right->dim==-1)

    {

        TREE tree=splitDataSet(dataSet,root->dim,root->value);

        for(int i=0;i<tree.first.size();i++) errNoMerga+=(tree.first[i].y-root->Left->value)*(tree.first[i].y-root->Left->value);

        for(int i=0;i<tree.second.size();i++) errNoMerga+=(tree.second[i].y-root->Right->value)*(tree.second[i].y-root->Right->value);

        double mergaMean=(root->Left->value+root->Right->value)/2;

        for(int i=0;i<dataSet.size();i++) errMerga+=(dataSet[i].y-mergaMean)*(dataSet[i].y-mergaMean);

        if(errMerga<errNoMerga) {/*cout<<"Merga"<<endl;*/return new RegTree(-1,mergaMean);}

        else return root;

    }

    return root;

}

int ccnt=0;

void displayTree(RegTree *root)

{

    if(root->Left->dim!=-1) displayTree(root->Left);

    else {printf("Leaf:%.2lf\n",root->Left->value);ccnt++;}

    if(root->Right->dim!=-1) displayTree(root->Right);

    else {printf("Leaf:%.2lf\n",root->Right->value);ccnt++;}

}

double forcast(RegTree *root,Data data)

{

    if(root->dim==-1) return root->value; //in case the super root is a leaf

    if(data.feature[root->dim]>root->value)

    {

        if(root->Left->dim!=-1) return forcast(root->Left,data);

        else return root->Left->value;

    }

    else

    {

        if(root->Right->dim!=-1) return forcast(root->Right,data);

        else return root->Right->value;

    }

}

void forcastAll(RegTree *root,vector<Data> dataSet)

{

    for(int i=0;i<dataSet.size();i++)

    {

        double y=forcast(root,dataSet[i]);

        printf("origin:%.2lf forcast:%.2lf\n",dataSet[i].y,y);

    }

}

int main()

{

    read();

    RegTree *root=buildTree(dataSet);

    root=prune(root,addSet);

    forcastAll(root,testSet);

}
回归树

 

你可能感兴趣的:(ca)