example_ensembledecisiontree.cpp

 
#include <iostream>
#include <Pds/Ra>
#include <Pds/Ml>
    
int main(void)
{
    Pds::Octave::XLabel="x_1";
    Pds::Octave::YLabel="x_2";
    Pds::Vector Yp;
    Pds::IterationConf Conf;
    Conf.Show=true; Conf.SetMinError(1e-07);    Conf.SetAlpha(0.1);
    
    // Load training data
    unsigned int L=2000;
    Pds::Matrix X;
    Pds::Vector Y;
    Pds::DataSet::Classification::LoadDataYinYang(L,X,Y);
 
    Pds::Octave::Plot::PointsX2DY(X,Y,"testando.m","example_ensembledecisiontree_data_yinyang.png");
    
    unsigned int StopSamples=3;
    unsigned int StopDeep=64;
    
    // Training Decisiontree
    Pds::DecisionTree::Counter=0;
    Pds::DecisionTree Arbol(X,Y,0.8,StopSamples,StopDeep);
    std::cout<<"Pds::DecisionTree::Counter: "<<Pds::DecisionTree::Counter<<std::endl;
    
    // Predict training data
    Yp=Arbol.Predict(X);
    std::cout<<"Acc:"<<Pds::Accuracy(Y,Yp,0.5)<<"\n";
    
    std::cout<<Pds::Ra::Repeat("*",128)<<"\n\n";
    
    Pds::Octave::Plot::PointsX2DY(X,Yp,"testando.m","example_ensembledecisiontree_simple_yinyang.png");
    
    // Training Ensemble Decisiontree
    Pds::DecisionTree::Counter=0;
    Pds::EnsembleDecisionTree Arbol2(X,Y,16,0.5,0.8,StopSamples,StopDeep);
    std::cout<<"Pds::DecisionTree::Counter: "<<Pds::DecisionTree::Counter<<std::endl;
    
    // Predict testing data
    Yp=Arbol2.Predict(X);
    std::cout<<"Acc:"<<Pds::Accuracy(Y,Yp,0.5)<<"\n";
    
    Pds::Octave::Plot::PointsX2DY(X,Yp,"testando.m","example_ensembledecisiontree_ensemble_yinyang.png");
    
    return 0;
}