diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2015-03-08 17:47:33 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2015-03-08 17:47:33 +0800 |
| commit | f2d01e30f459818f0589e06839d38999aecfdc06 (patch) | |
| tree | 9530ac898c1d4cdecbb5194cbd76288e57f7f7b1 /model/ranksvm.h | |
| parent | 22882d7113c13cb1e00c59b54050f16ac1b7cc30 (diff) | |
| download | ranksvm-f2d01e30f459818f0589e06839d38999aecfdc06.tar.gz ranksvm-f2d01e30f459818f0589e06839d38999aecfdc06.tar.bz2 ranksvm-f2d01e30f459818f0589e06839d38999aecfdc06.zip | |
scaffolding
Diffstat (limited to 'model/ranksvm.h')
| -rw-r--r-- | model/ranksvm.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h index e7b7c4a..21fb30b 100644 --- a/model/ranksvm.h +++ b/model/ranksvm.h @@ -12,8 +12,10 @@ protected: Eigen::VectorXd model; int fsize; public: - 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 + virtual int train(DataSet &D, Labels &label)=0; + virtual int predict(DataSet &D, Eigen::MatrixXd &res)=0; + // TODO Not sure how to construct this + // Possible solution: generate a nxn matrix each row contains the sorted list of ranker result. int saveModel(const std::string fname); static RSVM* loadModel(const std::string fname); virtual std::string getName()=0; |