diff options
Diffstat (limited to 'model')
| -rw-r--r-- | model/ranksvm.cpp | 1 | ||||
| -rw-r--r-- | model/ranksvm.h | 22 | 
2 files changed, 23 insertions, 0 deletions
diff --git a/model/ranksvm.cpp b/model/ranksvm.cpp new file mode 100644 index 0000000..38fb70c --- /dev/null +++ b/model/ranksvm.cpp @@ -0,0 +1 @@ +#include"ranksvm.h"
\ No newline at end of file diff --git a/model/ranksvm.h b/model/ranksvm.h new file mode 100644 index 0000000..ba79c48 --- /dev/null +++ b/model/ranksvm.h @@ -0,0 +1,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
\ No newline at end of file  | 
