baserec.knn package

Submodules

baserec.knn.item_knn_cf_recommender module

@author: Maurizio Ferrari Dacrema & Ceshine Lee

class baserec.knn.item_knn_cf_recommender.ItemKNNCFRecommender(URM_train, verbose=True)

Bases: baserec.base.base_similarity_matrix_recommender.BaseItemSimilarityMatrixRecommender

ItemKNN recommender (based on item-item similarities)

FEATURE_WEIGHTING_VALUES = ['BM25', 'TF-IDF', 'none']

RECOMMENDER_NAME = 'ItemKNNCFRecommender'

fit(topK=50, shrink=100, similarity='cosine', normalize=True, feature_weighting='none', **similarity_args)

baserec.knn.user_knn_cf_recommender module

@author: Maurizio Ferrari Dacrema & Ceshine Lee

class baserec.knn.user_knn_cf_recommender.UserKNNCFRecommender(URM_train, verbose=True)

Bases: baserec.base.base_similarity_matrix_recommender.BaseUserSimilarityMatrixRecommender

UserKNN recommender (based on user-user similarities)

FEATURE_WEIGHTING_VALUES = ['BM25', 'TF-IDF', 'none']

RECOMMENDER_NAME = 'UserKNNCFRecommender'

fit(topK=50, shrink=100, similarity='cosine', normalize=True, feature_weighting='none', **similarity_args)

Module contents