1 files changed, 2 insertions, 2 deletions
diff --git a/model/ranksvm.h b/model/ranksvm.h
index 21fb30b..5217b56 100644
--- a/model/ranksvm.h
+++ b/model/ranksvm.h
@@ -13,7 +13,7 @@ protected:
     int fsize;
 public:
     virtual int train(DataSet &D, Labels &label)=0;
-    virtual int predict(DataSet &D, Eigen::MatrixXd &res)=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);
@@ -21,7 +21,7 @@ public:
     virtual std::string getName()=0;
     Eigen::MatrixXd getModel(){
         return model;};
-    int setModel(const Eigen::VectorXd &model);
+    int setModel(const Labels &model);
 };
 
 #endif
 \ No newline at end of file