example_decisiontree_export.cpp

 
#include <iostream>
#include <Pds/Ra>
#include <Pds/Ml>
    
int main(void)
{
    Pds::Vector Yp;
    Pds::IterationConf Conf;
    Conf.Show=false; Conf.SetMinError(1e-07);    Conf.SetAlpha(0.1);
    
    // Load training data
    unsigned int L=1000;
    Pds::Matrix X;
    Pds::Vector Y;
    Pds::DataSet::Classification::LoadDataBand2(L,X,Y);
    
    Pds::Octave::XLabel="x_1";
    Pds::Octave::YLabel="x_2";
    Pds::Octave::Plot::PointsX2DY(X,Y,"testando.m","example_decisiontree_export_data.png");
    
    // Creando un arbol
    Pds::DecisionTree::Counter=0;
    Pds::DecisionTree Arbol("k2means",Conf,X,Y,0.8,4);
    std::cout<<"Pds::DecisionTree::Counter: "<<Pds::DecisionTree::Counter<<std::endl;
 
    // Exportando un txt 
    Arbol.ExportFile("arbol.txt");
 
    // Importando un txt
    Pds::DecisionTree Arbol2;
    Arbol2.ImportFile("arbol.txt");
    Arbol2.ExportFile("arbol2.txt");
    
    // Predict import arbol with training data
    Yp=Arbol2.Predict(X);
    
    Pds::Octave::Plot::PointsX2DY(X,Yp,"testando.m","example_decisiontree_export_import.png");
    
    // Importando un txt
    Pds::DecisionTree Arbol3("arbol.txt");
    Arbol3.ExportFile("arbol3.txt");
    
    // Predict import arbol with training data
    Yp=Arbol3.Predict(X);
    
    Pds::Octave::Plot::PointsX2DY(X,Yp,"testando.m","example_decisiontree_export_create.png");
 
    // Exportando arbol en archivo en formato dot.
    Arbol.ExportDotFile("arbol.dot");
    Pds::Ra::System("dot -Tpng -o example_decisiontree_export_arbol.png arbol.dot");   
 
    return 0;
}