1 files changed, 11 insertions, 8 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h
index ba79c48..8993b87 100644
--- a/model/ranksvm.h
+++ b/model/ranksvm.h
@@ -4,19 +4,22 @@
 #include<Eigen/Dense>
 #include<string>
 #include"../tools/dataProvider.h"
+#include "../tools/easylogging++.h"
 
 class RSVM  //Virtual base class for all RSVM operations
 {
 protected:
-    Eigen::VectorXd* model;
+    Eigen::VectorXd model;
+    int fsize;
 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();
+    virtual int train(DataProvider D)=0;
+    virtual int predict(DataProvider D);
+    int saveModel(std::string fname);
+    static RSVM loadModel(std::string fname);
+    virtual std::string getName()=0;
+    Eigen::MatrixXd getModel(){
+        return model;};
+    void setModel(Eigen::VectorXd model);
 };
 
 #endif
 \ No newline at end of file