DSSMItemDataset

class rectools.dataset.torch_datasets.DSSMItemDataset(items: csr_matrix)[source]

Bases: DSSMItemDatasetBase

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 item features and then returns them as tensors.

This class is intended for internal usage or advanced users.

Inherited-members

Methods

from_dataset(dataset)