ImplicitItemKNNWrapperModel

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

Bases: ModelBase[ImplicitItemKNNWrapperModelConfig]

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

dumps()

Serialize model to bytes.

fit(dataset, *args, **kwargs)

Fit model.

fit_partial(dataset, *args, **kwargs)

Fit model.

from_config(config)

Create model from config.

from_params(params[, sep])

Create model from parameters.

get_config([mode, simple_types])

Return model config.

get_params([simple_types, sep])

Return model parameters.

load(f)

Load model from file.

loads(data)

Load model from bytes.

recommend(users, dataset, k, filter_viewed)

Recommend items for users.

recommend_to_items(target_items, dataset, k)

Recommend items for target items.

save(f)

Save model to file.

Attributes

recommends_for_cold

recommends_for_warm

require_recommend_context

Indicates whether recommendation context is required for predictions.
config_class

alias of ImplicitItemKNNWrapperModelConfig