torch_mist.utils.data.dataset

Module Contents

Classes

SampleDataset

All the operations on a read-only sequence.

DataFrameDataset

DistributionDataset

WrappedDataset

class torch_mist.utils.data.dataset.SampleDataset(samples: Dict[str, torch.Tensor])

Bases: Sequence, torch.utils.data.Dataset

All the operations on a read-only sequence.

Concrete subclasses must override __new__ or __init__, __getitem__, and __len__.

__getitem__(item)
__len__()
class torch_mist.utils.data.dataset.DataFrameDataset(df: pandas.DataFrame)

Bases: torch.utils.data.Dataset

__len__()
__getitem__(item)
class torch_mist.utils.data.dataset.DistributionDataset(joint_dist: distutils.dist.Distribution | torch_mist.distributions.JointDistribution, max_samples: int = 100000, split_dim: int = -1)

Bases: torch.utils.data.Dataset

__len__() → int
__getitem__(idx) → Dict[str, torch.Tensor] | torch.Tensor
class torch_mist.utils.data.dataset.WrappedDataset(dataset: torch.utils.data.Dataset, func: Callable)

Bases: torch.utils.data.Dataset

__getitem__(item) → Any
__len__()