mcsvm.h

#ifndef LIBVM_MCSVM_H_
#define LIBVM_MCSVM_H_

#include "utilities.h"
#include "kernel.h"

enum { EXACT, APPROX, BINARY };  // redopt_type

struct MCSVMParameter {
  struct KernelParameter *kernel_param;
  int redopt_type;  // reduced optimization type
  int cache_size; // in Mb
  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 *num_svs;
  int *sv_indices;
  double **tau;
  struct Node **svs;
};

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);
double PredictMCSVMMaxValue(const struct MCSVMModel *model, const struct Node *x);

int SaveMCSVMModel(std::ofstream &model_file, const struct MCSVMModel *model);
MCSVMModel *LoadMCSVMModel(std::ifstream &model_file);
void FreeMCSVMModel(struct MCSVMModel *model);

void FreeMCSVMParam(struct MCSVMParameter *param);
void InitMCSVMParam(struct MCSVMParameter *param);
const char *CheckMCSVMParameter(const struct MCSVMParameter *param);

#endif  // LIBVM_MCSVM_H_