ImplicitItemKNNWrapperModel

class rectools.models.implicit_knn.ImplicitItemKNNWrapperModel(model: ItemItemRecommender, verbose: int = 0)[source]

Bases: ModelBase

Wrapper for implicit.nearest_neighbours.ItemItemRecommender and its successors.

See https://github.com/benfred/implicit/blob/main/implicit/nearest_neighbours.py for details.

Parameters
  • model (ItemItemRecommender) – Base model that will be used.

  • verbose (int, default 0) – Degree of verbose output. If 0, no output will be provided.

Inherited-members

Methods

fit(dataset, *args, **kwargs)

Fit model.

recommend(users, dataset, k, filter_viewed)

Recommend items for users.

recommend_to_items(target_items, dataset, k)

Recommend items for target items.

Attributes

recommends_for_cold

recommends_for_warm