1 files changed, 5 insertions, 5 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h
index 8993b87..fad790d 100644
--- a/model/ranksvm.h
+++ b/model/ranksvm.h
@@ -12,14 +12,14 @@ protected:
     Eigen::VectorXd model;
     int fsize;
 public:
-    virtual int train(DataProvider D)=0;
-    virtual int predict(DataProvider D);
-    int saveModel(std::string fname);
-    static RSVM loadModel(std::string fname);
+    virtual int train(DataProvider &D)=0;
+    virtual int predict(DataProvider &D)=0;
+    int saveModel(const std::string fname);
+    static RSVM* loadModel(const std::string fname);
     virtual std::string getName()=0;
     Eigen::MatrixXd getModel(){
         return model;};
-    void setModel(Eigen::VectorXd model);
+    int setModel(const Eigen::VectorXd &model);
 };
 
 #endif
 \ No newline at end of file