From e500bb4cdb32b13cc022b6dc5d221de7ad97a73e Mon Sep 17 00:00:00 2001 From: Joe Zhao Date: Sun, 8 Mar 2015 16:02:15 +0800 Subject: added commandline parser --- model/ranksvm.h | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'model/ranksvm.h') 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 -- cgit v1.2.3-70-g09d2