diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2015-04-10 20:39:00 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2015-04-10 20:39:00 +0800 |
| commit | 705f3731f4c49a75e2824d16622ff853634335c7 (patch) | |
| tree | 8c6a171615f27d0cb25484f72ccf1f84391eb9c3 /model/ranksvm.h | |
| parent | 01b523c7ce4eb5e692b0dcbec63efac0e8d1e2c7 (diff) | |
| download | ranksvm-705f3731f4c49a75e2824d16622ff853634335c7.tar.gz ranksvm-705f3731f4c49a75e2824d16622ff853634335c7.tar.bz2 ranksvm-705f3731f4c49a75e2824d16622ff853634335c7.zip | |
structuring input
Diffstat (limited to 'model/ranksvm.h')
| -rw-r--r-- | model/ranksvm.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h index b4ec7ce..e82b6be 100644 --- a/model/ranksvm.h +++ b/model/ranksvm.h @@ -3,6 +3,7 @@ #include<Eigen/Dense> #include<string> +#include<list> #include"../tools/dataProvider.h" #include "../tools/easylogging++.h" @@ -24,8 +25,8 @@ protected: SVMModel model; int fsize; public: - virtual int train(DataSet &D, Labels &label)=0; - virtual int predict(DataSet &D, Labels &res)=0; + virtual int train(DataList &D)=0; + virtual int predict(DataList &D,std::list<double> &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); |