def testGetLetorDcg(self):
     ev = LetorNdcgEval()
     self.assertAlmostEqual(ev.get_dcg([0, 0, 1, 0]), 0.6309298)
     self.assertEqual(ev.get_dcg([0, 1, 0, 0]), 1)
     self.assertEqual(ev.get_dcg([0, 0, 0, 0]), 0)
Esempio n. 2
0
 def testGetLetorDcg(self):
     ev = LetorNdcgEval()
     self.assertAlmostEqual(ev.get_dcg([0, 0, 1, 0]), 0.6309298)
     self.assertEqual(ev.get_dcg([0, 1, 0, 0]), 1)
     self.assertEqual(ev.get_dcg([0, 0, 0, 0]), 0)