PureSVDModel

class rectools.models.pure_svd.PureSVDModel(factors: int = 10, verbose: int = 0)

Bases: VectorModel

PureSVD matrix factorization model.

See https://dl.acm.org/doi/10.1145/1864708.1864721

Parameters

• factors (int, default 10) – The number of latent factors to compute.

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

Inherited-members

Methods

fit(dataset, *args, **kwargs)	Fit model.
get_vectors()	Return user and item vector representations from fitted model.
recommend(users, dataset, k, filter_viewed)	Recommend items for users.
recommend_to_items(target_items, dataset, k)	Recommend items for target items.

Attributes

i2i_dist
n_threads
recommends_for_cold
recommends_for_warm
u2i_dist

get_vectors() → Tuple[ndarray, ndarray]

Return user and item vector representations from fitted model.

Returns

User and item embeddings. Shapes are (n_users, n_factors) and (n_items, n_factors).

Return type

(np.ndarray, np.ndarray)