diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2015-03-11 00:55:41 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2015-03-11 00:55:41 +0800 |
| commit | f6c22b46449fa77f90e319e4b159ccb6c2a5732b (patch) | |
| tree | bc38dde03ffa5cb3c2fea9f5fefff0b990de405b /model/ranksvm.h | |
| parent | 3d204f5fe4614624ca342090feecbfe4df188d9d (diff) | |
| download | ranksvm-f6c22b46449fa77f90e319e4b159ccb6c2a5732b.tar.gz ranksvm-f6c22b46449fa77f90e319e4b159ccb6c2a5732b.tar.bz2 ranksvm-f6c22b46449fa77f90e319e4b159ccb6c2a5732b.zip | |
restructure, changed label type
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 21fb30b..5217b56 100644 --- a/model/ranksvm.h +++ b/model/ranksvm.h @@ -13,7 +13,7 @@ protected: int fsize; public: virtual int train(DataSet &D, Labels &label)=0; - virtual int predict(DataSet &D, Eigen::MatrixXd &res)=0; + virtual int predict(DataSet &D, Labels &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); @@ -21,7 +21,7 @@ public: virtual std::string getName()=0; Eigen::MatrixXd getModel(){ return model;}; - int setModel(const Eigen::VectorXd &model); + int setModel(const Labels &model); }; #endif
\ No newline at end of file |