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