torch_mist.utils.data.loader

Module Contents

Classes

SameAttributeDataLoader

Functions

sample_same_attributes(→ torch.utils.data.DataLoader)
sample_same_value(→ torch.utils.data.DataLoader)

class torch_mist.utils.data.loader.SameAttributeDataLoader(dataset: torch.utils.data.Dataset, attributes: torch.Tensor | numpy.ndarray, batch_size: int, neg_samples: int, **kwargs)

Bases: torch.utils.data.DataLoader

torch_mist.utils.data.loader.sample_same_attributes(dataloader: torch.utils.data.DataLoader, attributes: torch.Tensor | numpy.ndarray, neg_samples: int) → torch.utils.data.DataLoader

torch_mist.utils.data.loader.sample_same_value(dataloader: torch.utils.data.DataLoader, func: Callable[[Any], Any], neg_samples: int) → torch.utils.data.DataLoader