diff options
Diffstat (limited to 'model')
| -rw-r--r-- | model/ranksvm.cpp | 3 | ||||
| -rw-r--r-- | model/ranksvm.h | 4 |
2 files changed, 3 insertions, 4 deletions
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; |