elliot.recommender.latent_factor_models.Slim package

Submodules

elliot.recommender.latent_factor_models.Slim.slim module

Module description:

class elliot.recommender.latent_factor_models.Slim.slim.Slim(data, config, params, *args, **kwargs)[source]

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

Sparse Linear Methods

For further details, please refer to the paper

Parameters
  • l1_ratio

  • alpha

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

models:
  Slim:
    meta:
      save_recs: True
    epochs: 10
    l1_ratio: 0.001
    alpha: 0.001
get_recommendations(k: int = 100)[source]
property name
predict(u: int, i: int)[source]

Get prediction on the user item pair.

Returns

A single float vaue.

restore_weights()[source]
train()[source]

elliot.recommender.latent_factor_models.Slim.slim_model module

Module description:

class elliot.recommender.latent_factor_models.Slim.slim_model.SlimModel(data, num_users, num_items, l1_ratio, alpha, epochs)[source]

Bases: object

get_model_state()[source]
get_user_recs(user, k=100)[source]
predict(u, i)[source]
set_model_state(saving_dict)[source]
train(verbose)[source]

Module contents

Module description: