1 files changed, 4 insertions, 2 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h
index e7b7c4a..21fb30b 100644
--- a/model/ranksvm.h
+++ b/model/ranksvm.h
@@ -12,8 +12,10 @@ 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
+    virtual int train(DataSet &D, Labels &label)=0;
+    virtual int predict(DataSet &D, Eigen::MatrixXd &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;