Python LetorNdcgEval.get_ndcg 예제들

예제 #1

파일: test.py 프로젝트: anukat2015/AIR

 def testGetLetorNdcg(self):
     ev = LetorNdcgEval()
     self.assertAlmostEqual(ev.get_ndcg([0, 0, 1, 0], [1, 0, 0, 0]),
         0.6309298)
     self.assertEqual(ev.get_ndcg([0, 1, 0, 0], [1, 0, 0, 0]), 1)
     self.assertEqual(ev.get_ndcg([0, 0, 0, 0], [1, 0, 0, 0]), 0)

예제 #2

 def testGetLetorNdcg(self):
     ev = LetorNdcgEval()
     self.assertAlmostEqual(ev.get_ndcg([0, 0, 1, 0], [1, 0, 0, 0]),
                            0.6309298)
     self.assertEqual(ev.get_ndcg([0, 1, 0, 0], [1, 0, 0, 0]), 1)
     self.assertEqual(ev.get_ndcg([0, 0, 0, 0], [1, 0, 0, 0]), 0)