ott.neural.datasets.OTDataset#
- class ott.neural.datasets.OTDataset(src_data, tgt_data, src_conditions=None, tgt_conditions=None, is_aligned=False, seed=None)[source]#
Dataset for optimal transport problems.
- Parameters:
src_data (
OTData) – Samples from the source distribution.tgt_data (
OTData) – Samples from the target distribution.src_conditions (
Optional[Sequence[Any]]) – Conditions for the source data.tgt_conditions (
Optional[Sequence[Any]]) – Conditions for the target data.is_aligned (
bool) – Whether the samples from the source and the target data are paired. If yes, the source and the target conditions must match.seed (
Optional[int]) – Random seed used to match source and target when not aligned.
Methods
Attributes