diff options
Diffstat (limited to 'model/rankmisc.h')
-rw-r--r-- | model/rankmisc.h | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/model/rankmisc.h b/model/rankmisc.h new file mode 100644 index 0000000..2d2011d --- /dev/null +++ b/model/rankmisc.h @@ -0,0 +1,51 @@ +// +// Created by joe on 5/31/15. +// + +#ifndef RANKSVM_RANKMISC_H +#define RANKSVM_RANKMISC_H + +#include"ranksvm.h" + +class RSVML1:public RSVM +{ +public: + std::string getName() + { + return "L1"; + }; + virtual int train(RidList &D){LOG(FATAL)<< "NOT IMPLEMENTED"; return 0;}; + virtual int predict(RidList &D,std::vector<double> &res){ + res.clear(); + int n = D.getSize(); + Eigen::VectorXd one=Eigen::VectorXd::Ones(fsize); + for (int i=0;i<n;++i) + { + double r=D.getVecDot(i,one); + res.push_back(-r); + } + return 0; + }; +}; + +class RSVMBH:public RSVM +{ +public: + std::string getName() + { + return "BH"; + }; + virtual int train(RidList &D){LOG(FATAL)<< "NOT IMPLEMENTED"; return 0;}; + virtual int predict(RidList &D,std::vector<double> &res){ + res.clear(); + int n = D.getSize(); + for (int i=0;i<n;++i) + { + double r=D.getBha(i); + res.push_back(-r); + } + return 0; + }; +}; + +#endif //RANKSVM_RANKMISC_H |