c43310c317f67f13d7a80c1b1cdbc10abf28301b,matchzoo/metrics/rank_evaluations.py,rank_eval,ndcg,#rank_eval#Any#Any#Any#,64

Before Change

        c_g = sorted(c, key=lambda x:x[0], reverse=True)
        c_p = sorted(c, key=lambda x:x[1], reverse=True)
        idcg = [0. for i in range(k)]
        dcg = [0. for i in range(k)]
        for i, (g,p) in enumerate(c_g):
            if g > self.rel_threshold:
                idcg[i:] += (np.pow(2., g) - 1.) / np.log(2. + i)

After Change

        c_p = sorted(c, key=lambda x:x[1], reverse=True)
        //idcg = [0. for i in range(k)]
        idcg = np.zeros([k], dtype=np.float32)
        dcg = np.zeros([k], dtype=np.float32)
        //dcg = [0. for i in range(k)]
        for i, (g,p) in enumerate(c_g):
            if g > self.rel_threshold:
                idcg[i:] += (math.pow(2., g) - 1.) / math.log(2. + i)

Instances