VectorModel

class rectools.models.vector.VectorModel(verbose: int = 0, **kwargs: Any)[source]

Bases: ModelBase[ModelConfig_T]

Base class for models that represents users and items as vectors

Inherited-members

Parameters

  • verbose (int) –

  • kwargs (Any) –

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.

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

i2i_dist

recommends_for_cold

recommends_for_warm

u2i_dist