svmparam.h

//
// Created by jiashuai on 17-9-21.
//

#ifndef THUNDERSVM_SVMPARAM_H
#define THUNDERSVM_SVMPARAM_H

#include "thundersvm.h"

struct SvmParam {
    SvmParam() {
        svm_type = C_SVC;
        kernel_type = RBF;
        C = 1;
        gamma = 0;
        p = 0.1;
        epsilon = 0.001;
        nu = 0.5;
        probability = false;
        nr_weight = 0;
    }

    enum SVM_TYPE {
        C_SVC, NU_SVC, ONE_CLASS, EPSILON_SVR, NU_SVR
    };
    enum KERNEL_TYPE {
        LINEAR, POLY, RBF, SIGMOID/*, PRECOMPUTED*/
    };
    SVM_TYPE svm_type;
    KERNEL_TYPE kernel_type;

    float_type C;
    float_type gamma;
    float_type p;
    float_type nu;
    float_type epsilon;
    int degree;
    float_type coef0;
    int nr_weight;
    int *weight_label;
    float_type *weight;
    int probability;
    static const char *kernel_type_name[6];
    static const char *svm_type_name[6];
};
#endif //THUNDERSVM_SVMPARAM_H