VersatileClassificationTutorial-RF.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/Trees/RFClassifier.h>
#include <shark/Algorithms/Trainers/RFTrainer.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);

    RFClassifier<unsigned int> model;
    RFTrainer<unsigned int> trainer;

    trainer.train(model, traindata);

    auto 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;
}