model/ranksvm.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#ifndef RANKSVM_H
#define RANKSVM_H

#include<Eigen/Dense>
#include<string>
#include"../tools/dataProvider.h"

class RSVM  //Virtual base class for all RSVM operations
{
protected:
    Eigen::VectorXd* model;
public:
    virtual int train(DataProvider &D)=0;
    int test();
    int saveModel(string fname);
    static RSVM loadModel(string fname);
    string getName();
    Eigen::MatrixXd getModel();
    Eigen::MatrixXd setModel();
};

#endif