AbstractSVM.cpp

Go to the documentation of this file.

#include "AbstractSVM.h"

namespace damina
{

        AbstractSVM::AbstractSVM() {
                testSet = NULL;         
        }
        
        AbstractSVM::~AbstractSVM() {
                delete trainingSet;
                if (testSet != NULL) delete testSet;
        }


        void AbstractSVM::setKernel(int k) {
                this->params->kernel_type = k;
        }
        
        void AbstractSVM::setKernel(int k, double width) {
                this->params->kernel_type = k;
                this->params->gamma = width;
        }
        
        void AbstractSVM::setKernel(int k, int degree) {
                this->params->kernel_type = k;
                this->params->degree = degree;
        }
        
        
        void AbstractSVM::setKernel(int k, int degree, double width) {
                this->params->kernel_type = k;
                this->params->degree = degree;
                this->params->gamma = width;
        }
        
        void AbstractSVM::setKernelWidth(double w) {
                this->params->gamma = w;
        }
        
        void AbstractSVM::setKernelDegree(int deg) {
                this->params->degree = deg;
        }
        
        
        void AbstractSVM::setTolerance(double t) {
                this->params->eps = t;
        }
        
        void AbstractSVM::setCacheSize(int sizeMB) {
                this->params->cache_size = sizeMB;
        }
        
        void AbstractSVM::setTestSet(DataSet *test) {
                this->testSet = test;
        }
        
        void AbstractSVM::setTrainingSet(DataSet *train) {
                this->trainingSet = train;
        }
        
        DataSet *AbstractSVM::getTestSet() {
                return this->testSet;
        }
        
        
        DataSet *AbstractSVM::getTrainingSet() {
                return this->trainingSet;
        }
        
        
        int AbstractSVM::getKernel() {
                return this->params->kernel_type;
        }
        
        
        int AbstractSVM::getKernelDegree() {
                return this->params->degree;
        }
        
        double AbstractSVM::getKernelWidth() {
                return this->params->gamma;
        }
        
        
        double AbstractSVM::getTolerance() {
                return this->params->eps;
        }
        
        double AbstractSVM::getCacheSize() {
                return this->params->cache_size;
        }
        
        
        void AbstractSVM::enableShrinkingHeuristics() {
                this->params->shrinking = 1;
        }
        
        
        void AbstractSVM::disableShrinkingHeuristics() {
                this->params->shrinking = 0;
        }
        
        bool AbstractSVM::isShrinkingHeuristicsEnabled() {
                if (this->params->shrinking)
                        return true;
                        
                return false;
        }

        struct svm_parameter *AbstractSVM::getParameters() {
                return params;
        }
}

---