VersatileClassificationTutorial-LDA.cpp

Go to the documentation of this file.

#include <shark/Data/Dataset.h>
#include <shark/Data/Csv.h>
#include <shark/ObjectiveFunctions/Loss/ZeroOneLoss.h>

#include <shark/Models/LinearClassifier.h>
#include <shark/Algorithms/Trainers/LDA.h>


using namespace shark;

int main()
{
    // Load data, use 70% for training and 30% for testing.
    // The path is hard coded; make sure to invoke the executable
    // from a place where the data file can be found. It is located
    // under [shark]/examples/Supervised/data.
    ClassificationDataset traindata, testdata;
    importCSV(traindata, "data/quickstartData.csv", LAST_COLUMN, ' ');
    testdata = splitAtElement(traindata, 70 * traindata.numberOfElements() / 100);

    // TODO: define a model and a trainer

    LinearClassifier<> model;
    LDA trainer;

    trainer.train(model, traindata);

    Data<unsigned int> prediction = model(testdata.inputs());

    ZeroOneLoss<unsigned int> loss;
    double error_rate = loss(testdata.labels(), prediction);

    std::cout << "model: " << model.name() << std::endl
        << "trainer: " << trainer.name() << std::endl
        << "test error rate: " << error_rate << std::endl;
}
//