get_users_vectors

rectools.models.implicit_bpr.get_users_vectors(model: Union[BayesianPersonalizedRanking, BayesianPersonalizedRanking]) ndarray[source]

Get user vectors from BPR model as a numpy array.

Parameters

model (BayesianPersonalizedRanking) – Fitted BPR model. Can be CPU or GPU model

Returns

User vectors.

Return type

np.ndarray