From b44da2a2ab11425961014a39994484c92626ce58 Mon Sep 17 00:00:00 2001
From: Joe Zhao <ztuowen@gmail.com>
Date: Tue, 2 Jun 2015 18:10:33 +0800
Subject: misc

---
 model/rankaccu.cpp | 19 +++++++++++++++++++
 model/rankaccu.h   |  2 ++
 2 files changed, 21 insertions(+)

(limited to 'model')

diff --git a/model/rankaccu.cpp b/model/rankaccu.cpp
index 8404abf..0f55e26 100644
--- a/model/rankaccu.cpp
+++ b/model/rankaccu.cpp
@@ -128,4 +128,23 @@ void rank_CMC(RidList &D,const std::vector<double> pred,CMC & cmc) {
                 break; // account only for the first match;
             }
     }
+}
+
+void rank_pair(RidList &D,const vector<double> pred,vector<double> &pair)
+{
+    int n =D.getSize(),q=D.getqSize();
+    pair.clear();
+    for (int i=0;i<n;i+=q)
+    {
+        int corr=0;
+        for (int j=0;j<q;++j)
+            if (D.getL(i+j)>0) {
+                corr = j;
+                break;
+            }
+
+        for (int j=0;j<q;++j)
+            if (j!=corr)
+                pair.push_back(pred[i+corr]-pred[i+j]);
+    }
 }
\ No newline at end of file
diff --git a/model/rankaccu.h b/model/rankaccu.h
index e8da882..8cac56c 100644
--- a/model/rankaccu.h
+++ b/model/rankaccu.h
@@ -45,4 +45,6 @@ void rank_CMC(RidList &D,const std::vector<double> pred,CMC & cmc);
 
 void rank_accu(RidList &D,const std::vector<double> pred);
 
+void rank_pair(RidList &D,const std::vector<double> pred,std::vector<double> &pair);
+
 #endif //RANKSVM_RANKACCU_H
-- 
cgit v1.2.3-70-g09d2