Exemplo n.º 1
0
    def test_calculate_precision(self):

        expected_precision = 0.
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 1, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 1./3
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 3, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 3./5
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 5, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 3./7
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 7, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 0.
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_2, predicted_ratings_2, 1, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 2./3
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_2, predicted_ratings_2, 3, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 2./3
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_2, predicted_ratings_2, 5, 4.0)
        self.assertEqual(expected_precision, actual_precision)
Exemplo n.º 2
0
    def test_calculate_precision(self):

        expected_precision = 0.
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 1, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 1. / 3
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 3, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 3. / 5
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 5, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 3. / 7
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_1, predicted_ratings_1, 7, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 0.
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_2, predicted_ratings_2, 1, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 2. / 3
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_2, predicted_ratings_2, 3, 4.0)
        self.assertEqual(expected_precision, actual_precision)

        expected_precision = 2. / 3
        actual_precision = precision_in_top_n.calculate_precision(
            known_ratings_2, predicted_ratings_2, 5, 4.0)
        self.assertEqual(expected_precision, actual_precision)
Exemplo n.º 3
0
    def test_calculate_precision_none(self):

        expected_precision = None
        actual_precision = precision_in_top_n.calculate_precision([], [], 5,
                                                                  4.0)
        self.assertEqual(expected_precision, actual_precision)
Exemplo n.º 4
0
    def test_calculate_precision_none(self):

        expected_precision = None
        actual_precision = precision_in_top_n.calculate_precision([], [], 5, 4.0)
        self.assertEqual(expected_precision, actual_precision)