fscore

1 files changed, 72 insertions, 0 deletions

diff --git a/model/rankaccu.h b/model/rankaccu.h
index 8cac56c..a5866da 100644
--- a/model/rankaccu.h
+++ b/model/rankaccu.h
@@ -41,6 +41,78 @@ public:
     }
 };
 
+class Fscore
+{
+private:
+    std::vector<double> pos,neg;
+    std::vector<double> apos,aneg;
+    int cpos,cneg;
+    int f;
+public:
+    void clear() {
+        cpos=0;cneg=0;pos.clear();neg.clear();apos.clear();aneg.clear();
+    }
+    Fscore(){clear();}
+    void init(int fsize) {
+        f=fsize;pos.resize(fsize);neg.resize(fsize);apos.resize(fsize);aneg.resize(fsize);
+        for (int i=0;i<fsize;++i) apos[i]=aneg[i]=pos[i]=neg[i]=0;
+    }
+    void firstPass(RidList &rid,int x){
+        Eigen::VectorXd vec = rid.getVec(x);
+        std::vector<double> *p;
+        if (rid.getL(x)>0) {
+            p=&apos;
+            cpos+=1;
+        }
+        else {
+            p=&aneg;
+            cneg+=1;
+        }
+        for (int i=0;i<f;++i)
+            (*p)[i]+=vec(i);
+    }
+    void calAvg(){
+        for (int i=0;i<f;++i)
+            apos[i]/=cpos;
+        for (int i=0;i<f;++i)
+            aneg[i]/=cneg;
+    }
+    void secondPass(RidList &rid,int x){
+        Eigen::VectorXd vec = rid.getVec(x);
+        std::vector<double> *p,*a;
+        if (rid.getL(x)>0) {
+            p=&pos;
+            a=&apos;
+        }
+        else {
+            p=&neg;
+            a=&aneg;
+        }
+        for (int i=0;i<f;++i)
+            (*p)[i]+=(vec(i)-(*a)[i])*(vec(i)-(*a)[i]);
+    }
+    std::vector<double> getFscore(){
+        std::vector<double> res;
+        res.reserve(f);
+        for (int i=0;i<f;++i)
+        {
+            double avg;
+            avg = (cpos*apos[i]+cneg*aneg[i])/(cpos+cneg);
+            res[i] = (apos[i]-avg)*(apos[i]-avg)+(aneg[i]-avg)*(aneg[i]-avg);
+            res[i] /= pos[i]/(cpos-1)+neg[i]/(cneg-1);
+        }
+        return res;
+    }
+    void audit(RidList &rid){
+        init(rid.getfSize());
+        for (int i=0;i<rid.getSize();++i)
+            firstPass(rid,i);
+        calAvg();
+        for (int i=0;i<rid.getSize();++i)
+            secondPass(rid,i);
+    }
+};
+
 void rank_CMC(RidList &D,const std::vector<double> pred,CMC & cmc);
 
 void rank_accu(RidList &D,const std::vector<double> pred);