Python TopNEvaluator.update_num_hits Exemples

Exemple #1

    def test_calculate_recall(self):
        N = 5
        top_n_evaluator = TopNEvaluator([], [], N)

        top_n_list = ['I1', 'I2', 'I3', 'I4', 'I5']
        item = 'I3'
        top_n_evaluator.update_num_hits(top_n_list, item)

        top_n_list = ['I1', 'I2', 'I3', 'I4', 'I5']
        item = 'I6'
        top_n_evaluator.update_num_hits(top_n_list, item)

        top_n_list = ['I1', 'I6', 'I3', 'I4', 'I5']
        item = 'I4'
        top_n_evaluator.update_num_hits(top_n_list, item)

        self.assertEqual(2.0 / 3, top_n_evaluator.calculate_recall())

Exemple #2

Fichier : test_top_n_evaluator.py Projet : bachlog/yelp

    def test_calculate_recall(self):
        N = 5
        top_n_evaluator = TopNEvaluator([], [], N)

        top_n_list = ["I1", "I2", "I3", "I4", "I5"]
        item = "I3"
        top_n_evaluator.update_num_hits(top_n_list, item)

        top_n_list = ["I1", "I2", "I3", "I4", "I5"]
        item = "I6"
        top_n_evaluator.update_num_hits(top_n_list, item)

        top_n_list = ["I1", "I6", "I3", "I4", "I5"]
        item = "I4"
        top_n_evaluator.update_num_hits(top_n_list, item)

        self.assertEqual(2.0 / 3, top_n_evaluator.calculate_recall())

Exemple #3

    def test_update_num_hits(self):

        top_n_evaluator = TopNEvaluator([], [])
        self.assertEqual(0, top_n_evaluator.num_generic_hits)
        self.assertEqual(0, top_n_evaluator.num_generic_misses)

        top_n_list = ['I1', 'I2', 'I3', 'I4', 'I5']
        item = 'I3'
        top_n_evaluator.update_num_hits(top_n_list, item)
        self.assertEqual(1, top_n_evaluator.num_generic_hits)
        self.assertEqual(0, top_n_evaluator.num_generic_misses)

        top_n_list = ['I1', 'I2', 'I3', 'I4', 'I5']
        item = 'I6'
        top_n_evaluator.update_num_hits(top_n_list, item)
        self.assertEqual(1, top_n_evaluator.num_generic_hits)
        self.assertEqual(1, top_n_evaluator.num_generic_misses)

        top_n_list = ['I1', 'I6', 'I3', 'I4', 'I5']
        item = 'I4'
        top_n_evaluator.update_num_hits(top_n_list, item)
        self.assertEqual(2, top_n_evaluator.num_generic_hits)
        self.assertEqual(1, top_n_evaluator.num_generic_misses)

Exemple #4

Fichier : test_top_n_evaluator.py Projet : bachlog/yelp

    def test_update_num_hits(self):

        top_n_evaluator = TopNEvaluator([], [])
        self.assertEqual(0, top_n_evaluator.n_hit)
        self.assertEqual(0, top_n_evaluator.n_miss)

        top_n_list = ["I1", "I2", "I3", "I4", "I5"]
        item = "I3"
        top_n_evaluator.update_num_hits(top_n_list, item)
        self.assertEqual(1, top_n_evaluator.n_hit)
        self.assertEqual(0, top_n_evaluator.n_miss)

        top_n_list = ["I1", "I2", "I3", "I4", "I5"]
        item = "I6"
        top_n_evaluator.update_num_hits(top_n_list, item)
        self.assertEqual(1, top_n_evaluator.n_hit)
        self.assertEqual(1, top_n_evaluator.n_miss)

        top_n_list = ["I1", "I6", "I3", "I4", "I5"]
        item = "I4"
        top_n_evaluator.update_num_hits(top_n_list, item)
        self.assertEqual(2, top_n_evaluator.n_hit)
        self.assertEqual(1, top_n_evaluator.n_miss)