From 705f3731f4c49a75e2824d16622ff853634335c7 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Fri, 10 Apr 2015 20:39:00 +0800
Subject: structuring input

---
 model/ranksvm.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'model/ranksvm.h')

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);
-- 
cgit v1.2.3-70-g09d2