-
Notifications
You must be signed in to change notification settings - Fork 1
/
mcsvm.h
59 lines (50 loc) · 1.85 KB
/
mcsvm.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#ifndef LIBVM_MCSVM_H_
#define LIBVM_MCSVM_H_
#include "utilities.h"
enum { EXACT, APPROX, BINARY }; // redopt_type
enum { LINEAR, POLY, RBF, SIGMOID, PRECOMPUTED }; // kernel_type
struct MCSVMParameter {
int redopt_type; // reduced optimization type
int kernel_type;
int save_model;
int load_model;
int num_folds;
int probability;
int cache_size; // in Mb
int degree; // for poly
double gamma; // for poly/rbf/sigmoid
double coef0; // for poly/sigmoid
double beta;
double epsilon;
double epsilon0;
double delta;
};
struct MCSVMModel {
struct MCSVMParameter param;
int num_ex;
int num_classes; // number of classes (k)
int total_sv;
int *labels;
int *votes_weight;
int *num_svs;
int *sv_indices;
double *probA;
double *probB;
double **tau;
struct Node **svs;
};
void TrainProbMCSVM(const struct Problem *prob, const struct MCSVMParameter *param, struct MCSVMModel *model);
MCSVMModel *TrainMCSVM(const struct Problem *prob, const struct MCSVMParameter *param);
double *PredictMCSVMValues(const struct MCSVMModel *model, const struct Node *x);
int PredictMCSVM(const struct MCSVMModel *model, const struct Node *x, int *num_max_sim_score_ret);
int PredictProbMCSVM(const MCSVMModel *model, const Node *x, double *prob_estimates);
void CrossValidation(const struct Problem *prob, const struct MCSVMParameter *param, struct ErrStatistics *errors, int *predict_labels, double *probs, double *brier, double *logloss);
int SaveMCSVMModel(const char *file_name, const struct MCSVMModel *model);
MCSVMModel *LoadMCSVMModel(const char *file_name);
void FreeMCSVMModel(struct MCSVMModel *model);
void FreeMCSVMParam(struct MCSVMParameter *param);
void InitMCSVMParam(struct MCSVMParameter *param);
const char *CheckMCSVMParameter(const struct MCSVMParameter *param);
void SetPrintNull();
void SetPrintCout();
#endif // LIBVM_MCSVM_H_