diff options
Diffstat (limited to 'model/ranksvm.h')
-rw-r--r-- | model/ranksvm.h | 22 |
1 files changed, 22 insertions, 0 deletions
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 |