elliot.recommender.knn.attribute_item_knn package

Submodules

elliot.recommender.knn.attribute_item_knn.attribute_item_knn module

Module description:

class elliot.recommender.knn.attribute_item_knn.attribute_item_knn.AttributeItemKNN(data, config, params, *args, **kwargs)

Bases: elliot.recommender.recommender_utils_mixin.RecMixin, elliot.recommender.base_recommender_model.BaseRecommenderModel

Attribute Item-kNN proposed in MyMediaLite Recommender System Library

For further details, please refer to the paper

Parameters

• neighbors – Number of item neighbors

• similarity – Similarity function

To include the recommendation model, add it to the config file adopting the following pattern:

models:
  AttributeItemKNN:
    meta:
      save_recs: True
    neighbors: 40
    similarity: cosine

build_feature_sparse()

get_recommendations(k: int = 10)

get_single_recommendation(mask, k, *args)

property name

train()

elliot.recommender.knn.attribute_item_knn.attribute_item_knn_similarity module

class elliot.recommender.knn.attribute_item_knn.attribute_item_knn_similarity.Similarity(data, attribute_matrix, num_neighbors, similarity, implicit)

Bases: object

Simple kNN class

get_model_state()

get_user_recs(u, mask, k)

initialize()

This function initialize the data model

load_weights(path)

process_similarity(similarity)

save_weights(path)

set_model_state(saving_dict)

Module contents