svmClassification.h

Go to the documentation of this file.

 
#ifndef SVMCLASSIFICATION_H
#define SVMCLASSIFICATION_H
 
#include <vector>
#include "baseModel.h"
#include "../dependencies/libsvm/libsvm.h"
 
template<typename T>
class svmClassification final : public baseModel<T>
{
public:
    enum SVMType{ C_SVC = 0, NU_SVC, ONE_CLASS, EPSILON_SVR, NU_SVR };
    enum KernelType{ LINEAR_KERNEL = 0, POLY_KERNEL, RBF_KERNEL, SIGMOID_KERNEL, PRECOMPUTED_KERNEL };
    
    svmClassification(
                      KernelType kernelType = LINEAR_KERNEL,
                      SVMType svmType = C_SVC,
                      bool useScaling = true,
                      bool useNullRejection = false,
                      bool useAutoGamma = true,
                      float gamma = 0.1,
                      unsigned int degree = 3,
                      float coef0 = 0,
                      float nu = 0.5,
                      float C = 1,
                      bool useCrossValidation = false,
                      unsigned int kFoldValue = 10
                      );
    
    svmClassification(int numInputs);
    
    ~svmClassification();
    
    void train(const std::vector<trainingExampleTemplate<T> > &trainingSet) override;
 
    void train(const std::vector<trainingExampleTemplate<T> >& trainingSet, const std::size_t whichOutput) override;
    
    T run(const std::vector<T> &inputVector) override;
    
    void reset() override;
    
    bool init(KernelType kernelType,SVMType svmType,bool useScaling,bool useNullRejection,bool useAutoGamma,
              float gamma,
              unsigned int degree,
              float coef0,
              float nu,
              float C,
              bool useCrossValidation,
              unsigned int kFoldValue
              );
    
    size_t getNumInputs() const override;
    std::vector<size_t> getWhichInputs() const override;
        
#ifndef EMSCRIPTEN
    void getJSONDescription(Json::Value &currentModel) override;
#endif
    
private:
    bool problemSet;
    struct LIBSVM::svm_model *model;
    struct LIBSVM::svm_parameter param;
    struct LIBSVM::svm_problem problem;
    
    int numInputs;
    
    std::vector<double> inRanges;
    std::vector<double> inBases;
    
    bool trained;
};
 
#endif