example_kmeansnn.cpp

 
#include <iostream>
#include <Pds/Ra>
#include <Pds/Ml>
 
    
int main(void)
{
    Pds::ClassificationMetrics Metrics;
    Pds::IterationConf Conf;
    Conf.SetMaxIter(10000); Conf.Show=true;
    Pds::Octave::XLabel="x_1";
    Pds::Octave::YLabel="x_2";
    
    unsigned int K=5; // Number of clusters
    unsigned int L=1000; // Numer of 1s or 0s
    Pds::Matrix X;
    Pds::Vector Y,Yp;
    
    // Generating training data
    Pds::DataSet::Classification::LoadDataYinYang(L,X,Y);
    
    Pds::Octave::Plot::PointsX2DY(X,Y,"testando.m","example_kmeansnn_data.png");
 
    /********* Training  *********/
    
    // Create KmeansNn
    Pds::KmeansNn KmNn(Conf,X,Y,K);
    
    // Predict "training" data with Knn=7
    Yp=KmNn.Predict(X,7);
    
    // Metrics of training
    Metrics = Pds::ClassificationMetrics::Calculate(0.5,Yp,Y);
    Metrics.Print("\nMetrics of training:\n");
    
    Pds::Octave::Plot::PointsX2DY(X,Yp,"testando.m","example_kmeansnn_training.png");
    
    /********* Testing  *********/
    
    // Generating testing data
    Pds::DataSet::Classification::LoadDataYinYang(L,X,Y);
    
    // Predict testing data with Knn=7
    Yp=KmNn.Predict(X,7);
    
    // Metrics testing
    Metrics = Pds::ClassificationMetrics::Calculate(0.5,Yp,Y);
    Metrics.Print("\nMetrics of testing:\n");
    
    Pds::Octave::Plot::PointsX2DY(X,Yp,"testando.m","example_kmeansnn_testing.png");
    return 0;
}