diff options
Diffstat (limited to 'model/ranksvm.h')
| -rw-r--r-- | model/ranksvm.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h index e82b6be..aa5e1ca 100644 --- a/model/ranksvm.h +++ b/model/ranksvm.h @@ -26,7 +26,7 @@ protected: int fsize; public: virtual int train(DataList &D)=0; - virtual int predict(DataList &D,std::list<double> &res)=0; + virtual int predict(DataList &D,std::vector<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); |