Ejemplos de cum_gain en Python

Ejemplo n.º 1

Archivo: test_ndcg.py Proyecto: kmbnw/rank_metrics

    def test_param_stomping(self):
        testval_init = np.asarray([3, 2, 3, 0])
        testval = np.asarray([3, 2, 3, 0])

        x = ndcg.ndcg(testval, 4, False)
        self.assertEqual(testval_init.tolist(), testval.tolist())

        x = ndcg.idcg(testval, False)
        self.assertEqual(testval_init.tolist(), testval.tolist())

        x = ndcg.dcg(testval, False)
        self.assertEqual(testval_init.tolist(), testval.tolist())

        x = ndcg.cum_gain(testval)
        self.assertEqual(testval_init.tolist(), testval.tolist())

Ejemplo n.º 2

    def test_param_stomping(self):
        testval_init = np.asarray([3, 2, 3, 0])
        testval = np.asarray([3, 2, 3, 0])

        x = ndcg.ndcg(testval, 4, False)
        self.assertEqual(testval_init.tolist(), testval.tolist())

        x = ndcg.idcg(testval, False)
        self.assertEqual(testval_init.tolist(), testval.tolist())

        x = ndcg.dcg(testval, False)
        self.assertEqual(testval_init.tolist(), testval.tolist())

        x = ndcg.cum_gain(testval)
        self.assertEqual(testval_init.tolist(), testval.tolist())

Ejemplo n.º 3

Archivo: test_ndcg.py Proyecto: kmbnw/rank_metrics

 def test_cum_gain(self):
     # from wikipedia
     self.assertAlmostEqual(11.0, ndcg.cum_gain([3, 2, 3, 0, 1, 2]))
     # order is irrelevant
     self.assertAlmostEqual(11.0, ndcg.cum_gain([3, 1, 0, 3, 2, 2]))

Ejemplo n.º 4

 def test_cum_gain(self):
     # from wikipedia
     self.assertAlmostEqual(11.0, ndcg.cum_gain([3, 2, 3, 0, 1, 2]))
     # order is irrelevant
     self.assertAlmostEqual(11.0, ndcg.cum_gain([3, 1, 0, 3, 2, 2]))