OneClassSVM.cpp

Go to the documentation of this file.

#include "OneClassSVM.h"
#include <cstdio>

namespace damina
{

        OneClassSVM::OneClassSVM() {
                this->params = (struct svm_parameter *)malloc(sizeof(struct svm_parameter));
                this->params->svm_type = ONE_CLASS;
        }
        
        OneClassSVM::OneClassSVM(DataSet *train) {
                this->params = (struct svm_parameter *)malloc(sizeof(struct svm_parameter));
                this->params->svm_type = ONE_CLASS;
                this->trainingSetToProblem();
        }
        
        OneClassSVM::OneClassSVM(struct svm_problem *prob) {
                this->params = (struct svm_parameter *)malloc(sizeof(struct svm_parameter));
                this->params->svm_type = ONE_CLASS;
                this->problem = prob;
        }
        
        OneClassSVM::~OneClassSVM() {
                free(this->params);
                free(this->problem);
                free(this->model);
        }


        void OneClassSVM::setNU(double nu) {
                this->params->nu = nu;
        }
        
        double OneClassSVM::getNU() {
                return this->params->nu;
        }
        
        
        void OneClassSVM::learn() {
                this->model = svm_train(this->problem, this->params);
        }       
        
        void OneClassSVM::learn(DataSet *ts) {
                this->setTrainingSet(ts);
                this->trainingSetToProblem();
                this->learn();
        }
        
        void OneClassSVM::learn(struct svm_problem *p) {
                this->setProblem(p);
                this->learn();
        }
        
        void OneClassSVM::setProblem(struct svm_problem *p) {
                this->problem = p;
        }       
        
        
        struct svm_problem *OneClassSVM::getProblem() {
                return this->problem;
        }
        
        
        void OneClassSVM::setTrainingSet(DataSet *ts) {
                AbstractSVM::setTrainingSet(ts);
                this->trainingSetToProblem();
        }
        
        
        void OneClassSVM::classify() {
                //      TODO to implement
                // it acts on test set
        }
        
        void OneClassSVM::classify(DataSet *test) {
                // TODO to implement
                // it acts on test set
        }


        struct svm_model *OneClassSVM::getModel() {
                return this->model;
        }
        
        
        // private methods
        
        
        void OneClassSVM::trainingSetToProblem() {
                DataSet *ts = this->getTrainingSet();
                Point *temp;
                
                unsigned long int i, j;
                                        
                this->problem->l = ts->getSize();
                this->problem->y = (double *) malloc(this->problem->l * sizeof(double));
                this->problem->x = (struct svm_node **) malloc(this->problem->l * sizeof(struct svm_node*));
                
                // iteration on Points
                for (i = 0; i < ts->getSize(); i++) {
                        temp = ts->get(i);
                        
                        this->problem->x[i] = (struct svm_node*) malloc((temp->getSize() + 1) * sizeof(struct svm_node));
                        this->problem->y[i] = temp->getLabel();
                        
                        //iteration on Features
                        for (j = 0; j < temp->getSize(); j++) {
                                this->problem->x[i][j].index = temp->get(j)->getIndex();
                                this->problem->x[i][j].value = temp->get(j)->getValue();                                
                        }
                        
                        this->problem->x[i][j].index = -1;
                        this->problem->x[i][j].value = 0;
                }
        }
}

---