elliot.recommender.NN.attribute_item_knn package

Submodules

elliot.recommender.NN.attribute_item_knn.attribute_item_knn module

Module description:

class elliot.recommender.NN.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 = 100)

property name

restore_weights()

train()

elliot.recommender.NN.attribute_item_knn.attribute_item_knn_similarity module

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

Bases: object

Simple kNN class

compute_cosine(i_index, j_index)

compute_neighbors()

get_item_neighbors(item)

get_model_state()

get_transactions()

get_user_recs(u, k)

initialize()

This function initialize the data model

process_similarity(similarity)

static score_item(neighs, user_items)

set_model_state(saving_dict)

Module contents