diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2015-03-08 16:02:15 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2015-03-08 16:02:15 +0800 |
| commit | e500bb4cdb32b13cc022b6dc5d221de7ad97a73e (patch) | |
| tree | 643bcf86336437ccee6182fc6d19c92f33b7201a /model/ranksvm.h | |
| parent | 457024eedfaf6e08146038c8cb3034e590a81df6 (diff) | |
| download | ranksvm-e500bb4cdb32b13cc022b6dc5d221de7ad97a73e.tar.gz ranksvm-e500bb4cdb32b13cc022b6dc5d221de7ad97a73e.tar.bz2 ranksvm-e500bb4cdb32b13cc022b6dc5d221de7ad97a73e.zip | |
added commandline parser
Diffstat (limited to 'model/ranksvm.h')
| -rw-r--r-- | model/ranksvm.h | 10 |
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 |