From 22882d7113c13cb1e00c59b54050f16ac1b7cc30 Mon Sep 17 00:00:00 2001 From: Joe Zhao Date: Sun, 8 Mar 2015 16:42:53 +0800 Subject: migrating & scaffolding & reformatting Added more cmdline options --- model/ranksvm.cpp | 3 +-- model/ranksvm.h | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) (limited to 'model') diff --git a/model/ranksvm.cpp b/model/ranksvm.cpp index 58a097a..6294245 100644 --- a/model/ranksvm.cpp +++ b/model/ranksvm.cpp @@ -25,7 +25,6 @@ RSVM* RSVM::loadModel(const string fname){ RSVM* rsvm; - // TODO multiplex type if (type=="TN") rsvm = new RSVMTN(); @@ -39,7 +38,7 @@ RSVM* RSVM::loadModel(const string fname){ int RSVM::setModel(const Eigen::VectorXd &model) { if (model.cols()!=fsize) - LOG(FATAL) << "Feature size mismatch";; + LOG(FATAL) << "Feature size mismatch"; this->model=model; return 0; } \ No newline at end of file diff --git a/model/ranksvm.h b/model/ranksvm.h index fad790d..e7b7c4a 100644 --- a/model/ranksvm.h +++ b/model/ranksvm.h @@ -12,8 +12,8 @@ protected: Eigen::VectorXd model; int fsize; public: - virtual int train(DataProvider &D)=0; - virtual int predict(DataProvider &D)=0; + virtual int train(DataProvider &D)=0; // Dataprovider will have to provide label + virtual int predict(DataProvider &D)=0; // TODO Not sure how to construct this int saveModel(const std::string fname); static RSVM* loadModel(const std::string fname); virtual std::string getName()=0; -- cgit v1.2.3-70-g09d2