CMC auditing & using std in main

1 files changed, 31 insertions, 0 deletions

diff --git a/model/rankaccu.cpp b/model/rankaccu.cpp
index 069e245..1763e0f 100644
--- a/model/rankaccu.cpp
+++ b/model/rankaccu.cpp
@@ -112,4 +112,35 @@ int rank_accu(DataList &D,const vector<double> pred)
         ++i;
     }
     LOG(INFO)<<"over "<< cnt<< " queries. "<<"Average nDGC: "<< accu_nDCG/cnt<< " Average AP: "<<accu_AP/cnt;
+}
+
+int rank_CMC(DataList &D,const std::vector<double> pred,CMC & cmc) {
+    unsigned long n = D.getSize();
+    vector<int> orig_rank(n),pred_rank(n),C(n);
+    vector<double> orig(n);
+    int i,j;
+    for (i=0;i<D.getSize();++i)
+    {
+        orig_rank[i]=i;
+        pred_rank[i]=i;
+        orig[i]=D.getData()[i]->rank;
+    }
+    int cnt=0;
+    i=j=0;
+    while (i<D.getSize())
+    {
+        if ((i+1 == D.getSize())|| D.getData()[i]->qid!=D.getData()[i+1]->qid)
+        {
+            ranksort(j,i,pred_rank,pred,orig);
+            for (int k=j;k<=i;++k)
+                if (orig[pred_rank[k]]>0)
+                {
+                    cmc.addEntry(k-j);
+                    break; // account only for the first match;
+                }
+            j = i+1;
+            ++cnt;
+        }
+        ++i;
+    }
 }
\ No newline at end of file