Python WeightedSumRecommenderの例

コード例 #1

ファイル: test_weighted_sum_recommender.py プロジェクト: antoine-tran/yelp

    def test_predict_rating(self):

        recommender = WeightedSumRecommender(SingleSimilarityMatrixBuilder('cosine'))
        recommender.load(reviews)

        actual_rating_1 = 4.0
        self.assertEqual(actual_rating_1, recommender.predict_rating('A1', 1))
        actual_rating_2 = 2.0
        self.assertEqual(actual_rating_2, recommender.predict_rating('A1', 3))

コード例 #2

ファイル: test_precision_in_top_n.py プロジェクト: neostoic/yelp-1

    def test_calculate_recommender_precision(self):

        recommender = WeightedSumRecommender(
            SingleSimilarityMatrixBuilder('euclidean'))
        recommender.load(reviews_matrix_5)

コード例 #3

ファイル: test_precision_in_top_n.py プロジェクト: antoine-tran/yelp

    def test_calculate_recommender_precision(self):

        recommender = WeightedSumRecommender(SingleSimilarityMatrixBuilder('euclidean'))
        recommender.load(reviews_matrix_5)