DSSMUserDataset

class rectools.dataset.torch_datasets.DSSMUserDataset(users: csr_matrix, interactions: csr_matrix, keep_users: Optional[Sequence[int]] = None)[source]

Bases: DSSMUserDatasetBase

Torch dataset wrapper for rectools.dataset.dataset.Dataset. Implements torch.utils.data.Dataset for subsequent usage with torch.utils.data.DataLoader. Does the following: for a given index takes a row of user interactions, a row of user features and then returns them as tensors.

This class is intended for internal usage or advanced users.

Inherited-members

Methods

from_dataset(dataset[, keep_users])