diff options
Diffstat (limited to 'model/rankaccu.cpp')
| -rw-r--r-- | model/rankaccu.cpp | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/model/rankaccu.cpp b/model/rankaccu.cpp new file mode 100644 index 0000000..2e77eb6 --- /dev/null +++ b/model/rankaccu.cpp @@ -0,0 +1,69 @@ +// +// Created by joe on 4/12/15. +// + +#include "rankaccu.h" +#include "../tools/easylogging++.h" + +using namespace std; + +const double offset = 1; + +void ranksort(int l,int r,vector<int> &rank,const vector<double> &ref1,const vector<double> &ref2) +{ + int i=l,j=r,k; + double mid1=ref1[rank[(l+r)>>1]],mid2=ref2[rank[(l+r)>>1]]; + while (i<=j) + { + while (ref1[rank[i]]>mid1 || (ref1[rank[i]]==mid1 && ref2[rank[i]]>mid2)) ++i; + while (ref1[rank[j]]<mid1 || (ref1[rank[j]]==mid1 && ref2[rank[j]]<mid2)) --j; + if (i<=j) + { + k=rank[i]; + rank[i]=rank[j]; + rank[j]=k; + ++i; + --j; + } + } + if (j>l) + ranksort(l,j,rank,ref1,ref2); + if (i<r) + ranksort(i,r,rank,ref1,ref2); +} + +int rank_accu(DataList &D,const vector<double> pred) +{ + unsigned long n = D.getSize(); + vector<int> orig_rank(n),pred_rank(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; + double accu_nDCG=0; + i=j=0; + while (i<D.getSize()) + { + if ((i+1 == D.getSize())|| D.getData()[i]->qid!=D.getData()[i+1]->qid) + { + double Y=0,Z=0; + ranksort(j,i,orig_rank,orig,pred); + ranksort(j,i,pred_rank,pred,orig); + for (int k = j;k<=i;++k) + { + Z += (pow(2,offset+orig[orig_rank[k]]) - 1)/log2(2+k-j); + Y += (pow(2,offset+orig[pred_rank[k]]) - 1)/log2(2+k-j); + } + accu_nDCG+=Y/Z; + j = i+1; + ++cnt; + } + ++i; + } + LOG(INFO)<<"Average nDGC over "<< cnt<< " queries: "<< accu_nDCG/cnt; +}
\ No newline at end of file |