From 93b28277476de5d4fbfbf5ac236f9f619d482c46 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Mon, 8 Jun 2015 11:20:36 +0800
Subject: separation, HE

---
 model/rankmisc.h  | 20 ++++++++++++++++++++
 model/ranksvm.cpp |  2 ++
 2 files changed, 22 insertions(+)

(limited to 'model')

diff --git a/model/rankmisc.h b/model/rankmisc.h
index 2d2011d..33d21a4 100644
--- a/model/rankmisc.h
+++ b/model/rankmisc.h
@@ -48,4 +48,24 @@ public:
     };
 };
 
+class RSVMHE:public RSVM
+{
+public:
+    std::string getName()
+    {
+        return "HE";
+    };
+    virtual int train(RidList &D){LOG(FATAL)<< "NOT IMPLEMENTED"; return 0;};
+    virtual int predict(RidList &D,std::vector<double> &res){
+        res.clear();
+        int n = D.getSize();
+        for (int i=0;i<n;++i)
+        {
+            double r=D.getHell(i);
+            res.push_back(-r);
+        }
+        return 0;
+    };
+};
+
 #endif //RANKSVM_RANKMISC_H
diff --git a/model/ranksvm.cpp b/model/ranksvm.cpp
index ed78fbe..20d65e3 100644
--- a/model/ranksvm.cpp
+++ b/model/ranksvm.cpp
@@ -48,6 +48,8 @@ RSVM* RSVM::loadModel(const string fname){
         rsvm = new RSVML1();
     if (type=="BH")
         rsvm = new RSVMBH();
+    if (type=="HE")
+        rsvm = new RSVMHE();
 
     rsvm->fsize=fsize;
     SVMModel model;
-- 
cgit v1.2.3-70-g09d2