torch_datasets

Special datasets used in neural models.

Classes

DSSMItemDataset(items)	Torch dataset wrapper for rectools.dataset.dataset.Dataset.
DSSMItemDatasetBase(*args, **kwargs)	Base class for DSSM item datasets.
DSSMTrainDataset(items, users, interactions)	Torch dataset wrapper for rectools.dataset.dataset.Dataset.
DSSMTrainDatasetBase(*args, **kwargs)	Base class for DSSM training datasets.
DSSMUserDataset(users, interactions[, ...])	Torch dataset wrapper for rectools.dataset.dataset.Dataset.
DSSMUserDatasetBase(*args, **kwargs)	Base class for DSSM training datasets.