1 files changed, 3 insertions, 2 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h
index b4ec7ce..e82b6be 100644
--- a/model/ranksvm.h
+++ b/model/ranksvm.h
@@ -3,6 +3,7 @@
 
 #include<Eigen/Dense>
 #include<string>
+#include<list>
 #include"../tools/dataProvider.h"
 #include "../tools/easylogging++.h"
 
@@ -24,8 +25,8 @@ protected:
     SVMModel model;
     int fsize;
 public:
-    virtual int train(DataSet &D, Labels &label)=0;
-    virtual int predict(DataSet &D, Labels &res)=0;
+    virtual int train(DataList &D)=0;
+    virtual int predict(DataList &D,std::list<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);