#ifndef RANKSVM_H #define RANKSVM_H #include #include #include"../tools/dataProvider.h" #include "../tools/easylogging++.h" class RSVM //Virtual base class for all RSVM operations { protected: Eigen::VectorXd model; int fsize; public: virtual int train(DataSet &D, Labels &label)=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); static RSVM* loadModel(const std::string fname); virtual std::string getName()=0; Eigen::MatrixXd getModel(){ return model;}; int setModel(const Labels &model); }; #endif