summaryrefslogtreecommitdiff
path: root/model/ranksvm.h
diff options
context:
space:
mode:
Diffstat (limited to 'model/ranksvm.h')
-rw-r--r--model/ranksvm.h22
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