separation, HE

2 files changed, 22 insertions, 0 deletions

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;