#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(DataProvider &D)=0; // Dataprovider will have to provide label virtual int predict(DataProvider &D)=0; // TODO Not sure how to construct this 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 Eigen::VectorXd &model); }; #endif