Function TEST

src/test/test_oneclass_svc.cpp:7–30

#include <thundersvm/model/oneclass_svc.h>

TEST(OneClassSVCTest, train) {
    DataSet dataset;
    dataset.load_from_file(DATASET_DIR "test_dataset.txt");
    SvmParam param;
    param.gamma = 0.5;
    param.nu = 0.1;
    param.epsilon = 0.001;
    param.kernel_type = SvmParam::RBF;
    param.svm_type = SvmParam::ONE_CLASS;
    SvmModel *model = new OneClassSVC();
    model->train(dataset, param);
    model->save_to_file(DATASET_DIR "test_dataset.txt.model");
    SvmModel *new_model = new OneClassSVC();
    new_model->load_from_file(DATASET_DIR "test_dataset.txt.model");
    vector<float_type> predict_y = new_model->predict(dataset.instances(), 100);
    int n_pos = 0;
    for (unsigned i = 0; i < predict_y.size(); ++i) {
        if (predict_y[i] > 0)
            n_pos++;
    }
    EXPECT_EQ(n_pos, 135);
    delete model;
    delete new_model;
}