Index

_

__call__() (omniloader.collate.DynamicCollator method) (omniloader.schema.unify.SampleUnifier method) (omniloader.transforms.base.Compose method) (omniloader.transforms.base.Transform method) (omniloader.transforms.mix.MixupCollator method) __getitem__() (omniloader.data.datasets.DictTensorDataset method) (omniloader.data.datasets.HDF5Dataset method) (omniloader.data.npy.NpyFolderDataset method) (omniloader.loader.OmniLoader method) (omniloader.loader.SizedDataset method) __getstate__() (omniloader.data.datasets.HDF5Dataset method) __iter__() (omniloader.sampling.bucketing.LengthBucketBatchSampler method) (omniloader.sampling.bucketing.SizedSampler method) (omniloader.sampling.sampler.OmniSampler method)

__len__() (omniloader.data.datasets.DictTensorDataset method) (omniloader.data.datasets.HDF5Dataset method) (omniloader.data.npy.NpyFolderDataset method) (omniloader.loader.OmniLoader method) (omniloader.loader.SizedDataset method) (omniloader.sampling.bucketing.LengthBucketBatchSampler method) (omniloader.sampling.bucketing.SizedSampler method) (omniloader.sampling.sampler.OmniSampler method) (omniloader.sampling.strategies.MixingStrategy method) (omniloader.schema.spec.UnifiedSchema method) __post_init__() (omniloader.schema.spec.TensorSpec method) __str__() (omniloader.introspection.Report method) __version__ (in module omniloader)

A

ADAPTERS (in module omniloader.data.factory) AnnealedTemperatureStrategy (class in omniloader.sampling.strategies) apply() (omniloader.transforms.augment.FeatureDropout method) (omniloader.transforms.augment.FeatureMasking method) (omniloader.transforms.augment.GaussianNoise method) (omniloader.transforms.augment.SpanMasking method) (omniloader.transforms.augment.TimeWarp method) (omniloader.transforms.base.Compose method) (omniloader.transforms.base.Transform method) (omniloader.transforms.crop.CenterCrop method) (omniloader.transforms.crop.RandomCrop method) (omniloader.transforms.normalize.InstanceNormalize method) (omniloader.transforms.normalize.PerDatasetNormalize method)

B

broadcast_mask() (in module omniloader.transforms.base) build_collate() (omniloader.config.OmniConfig method) build_dataloader() (omniloader.config.OmniConfig method) build_dataset() (in module omniloader.data.factory) build_datasets() (in module omniloader.data.factory) (omniloader.config.OmniConfig method)

build_loader() (omniloader.config.OmniConfig method) build_parser() (in module omniloader.cli) build_strategy() (omniloader.config.OmniConfig method) build_subsample() (omniloader.config.OmniConfig method) build_transform() (in module omniloader.transforms) (omniloader.config.OmniConfig method)

C

CenterCrop (class in omniloader.transforms.crop) class_distributions (omniloader.introspection.Report attribute) class_histogram() (in module omniloader.sampling.weights) class_weights_for_loss() (in module omniloader.sampling.weights) class_weights_for_sampler() (in module omniloader.sampling.weights) COLLATES (in module omniloader.config)

Compose (class in omniloader.transforms.base) compute_dataset_stats() (in module omniloader.transforms.stats) compute_feature_stats() (in module omniloader.transforms.stats) compute_stats() (in module omniloader.transforms.stats) CONFIG_TEMPLATE (in module omniloader.templates) config_template_path() (in module omniloader.templates) coverage (omniloader.introspection.Report attribute)

D

DatasetSchema (class in omniloader.schema.spec) DEFAULT_NAMES (in module omniloader.data.splits) describe() (in module omniloader.introspection) (omniloader.loader.OmniLoader method)

DictTensorDataset (class in omniloader.data.datasets) draw() (omniloader.sampling.subsamplers.IndexPool method) DynamicCollator (class in omniloader.collate)

E

epoch_indices() (omniloader.sampling.strategies.MixingStrategy method) (omniloader.sampling.strategies.RoundRobinStrategy method)

EXHAUST (omniloader.sampling.subsamplers.ExhaustionPolicy attribute) ExhaustionPolicy (class in omniloader.sampling.subsamplers)

F

feature_keys (omniloader.schema.spec.UnifiedSchema property) FeatureDropout (class in omniloader.transforms.augment) FeatureMasking (class in omniloader.transforms.augment) fill_gaps_with_repeat() (in module omniloader.utils.padding) FixedWeightStrategy (class in omniloader.sampling.strategies)

FRESH (omniloader.sampling.subsamplers.ExhaustionPolicy attribute) from_dict() (omniloader.config.OmniConfig class method) from_file() (omniloader.config.OmniConfig class method) from_json() (omniloader.config.OmniConfig class method) from_yaml() (omniloader.config.OmniConfig class method)

G

GaussianNoise (class in omniloader.transforms.augment)

H

HDF5Dataset (class in omniloader.data.datasets)

I

IndexPool (class in omniloader.sampling.subsamplers)

InstanceNormalize (class in omniloader.transforms.normalize) is_sequence (omniloader.schema.spec.TensorSpec property)

K

keys (omniloader.schema.spec.DatasetSchema property) (omniloader.schema.spec.UnifiedSchema property)

L

LengthBucketBatchSampler (class in omniloader.sampling.bucketing) load_split_info() (in module omniloader.data.splits)

load_stats() (in module omniloader.transforms.stats) locate() (omniloader.loader.OmniLoader method)

M

main() (in module omniloader.cli) make_sampler() (omniloader.loader.OmniLoader method) mask_shape (omniloader.schema.spec.TensorSpec property) METADATA_KEYS (in module omniloader.schema.unify) MinMaxNormalize (class in omniloader.transforms.normalize) MixingStrategy (class in omniloader.sampling.strategies) MixupCollator (class in omniloader.transforms.mix) module omniloader omniloader.cli omniloader.collate omniloader.config omniloader.data omniloader.data.datasets omniloader.data.factory omniloader.data.npy omniloader.data.splits omniloader.integrations omniloader.integrations.lightning omniloader.introspection omniloader.loader omniloader.sampling omniloader.sampling.bucketing omniloader.sampling.sampler omniloader.sampling.strategies omniloader.sampling.subsamplers omniloader.sampling.weights omniloader.schema omniloader.schema.spec omniloader.schema.unify omniloader.templates omniloader.transforms omniloader.transforms.augment omniloader.transforms.base omniloader.transforms.crop omniloader.transforms.mix omniloader.transforms.normalize omniloader.transforms.stats omniloader.utils omniloader.utils.padding omniloader.utils.seeding

N

names (omniloader.introspection.Report attribute)

Normalize (class in omniloader.transforms.normalize) NpyFolderDataset (class in omniloader.data.npy)

O

OmniConfig (class in omniloader.config) OmniDataModule (class in omniloader.integrations.lightning) omniloader module OmniLoader (class in omniloader.loader) omniloader.cli module omniloader.collate module omniloader.config module omniloader.data module omniloader.data.datasets module omniloader.data.factory module omniloader.data.npy module omniloader.data.splits module omniloader.integrations module omniloader.integrations.lightning module omniloader.introspection module omniloader.loader module omniloader.sampling module omniloader.sampling.bucketing module omniloader.sampling.sampler module

omniloader.sampling.strategies module omniloader.sampling.subsamplers module omniloader.sampling.weights module omniloader.schema module omniloader.schema.spec module omniloader.schema.unify module omniloader.templates module omniloader.transforms module omniloader.transforms.augment module omniloader.transforms.base module omniloader.transforms.crop module omniloader.transforms.mix module omniloader.transforms.normalize module omniloader.transforms.stats module omniloader.utils module omniloader.utils.padding module omniloader.utils.seeding module OmniSampler (class in omniloader.sampling.sampler)

P

pad_or_crop_time_dim() (in module omniloader.utils.padding) PerDatasetNormalize (class in omniloader.transforms.normalize)

placeholder_mask() (omniloader.schema.spec.TensorSpec method) placeholder_value() (omniloader.schema.spec.TensorSpec method) ProportionalStrategy (class in omniloader.sampling.strategies)

R

RandomCrop (class in omniloader.transforms.crop) repeat_pad_time_dim() (in module omniloader.utils.padding) Report (class in omniloader.introspection) reset_for_worker() (omniloader.data.datasets.HDF5Dataset method)

resolve_dtype() (in module omniloader.schema.spec) resolve_keys() (in module omniloader.transforms.base) resolve_sequence_keys() (in module omniloader.transforms.base) RobustNormalize (class in omniloader.transforms.normalize) RoundRobinStrategy (class in omniloader.sampling.strategies)

S

Sample (in module omniloader.transforms.base) SampleUnifier (class in omniloader.schema.unify) save_split_info() (in module omniloader.data.splits) save_stats() (in module omniloader.transforms.stats) schema_from_dict() (in module omniloader.data.factory) seed_everything() (in module omniloader.config) seed_worker() (in module omniloader.utils.seeding) sequence_lengths() (omniloader.loader.OmniLoader method) set_epoch() (omniloader.integrations.lightning.OmniDataModule method) (omniloader.loader.OmniLoader method) (omniloader.sampling.sampler.OmniSampler method)

setup() (omniloader.integrations.lightning.OmniDataModule method) SizedDataset (class in omniloader.loader) SizedSampler (class in omniloader.sampling.bucketing) sizes (omniloader.introspection.Report attribute) SpanMasking (class in omniloader.transforms.augment) spec() (omniloader.schema.spec.UnifiedSchema method) specs (omniloader.schema.spec.DatasetSchema property) (omniloader.schema.spec.UnifiedSchema property) split_indices() (in module omniloader.data.splits) STRATEGIES (in module omniloader.config) SubsampleConfig (class in omniloader.sampling.strategies)

T

target_counts() (omniloader.sampling.strategies.AnnealedTemperatureStrategy method) (omniloader.sampling.strategies.FixedWeightStrategy method) (omniloader.sampling.strategies.MixingStrategy method) (omniloader.sampling.strategies.ProportionalStrategy method) (omniloader.sampling.strategies.RoundRobinStrategy method) (omniloader.sampling.strategies.TemperatureStrategy method) target_keys (omniloader.schema.spec.UnifiedSchema property) temperature_at() (omniloader.sampling.strategies.AnnealedTemperatureStrategy method) TemperatureStrategy (class in omniloader.sampling.strategies)

TensorSpec (class in omniloader.schema.spec) test_dataloader() (omniloader.integrations.lightning.OmniDataModule method) TimeWarp (class in omniloader.transforms.augment) to_dict() (omniloader.config.OmniConfig method) trailing_shape (omniloader.schema.spec.TensorSpec property) train_dataloader() (omniloader.integrations.lightning.OmniDataModule method) train_only (omniloader.transforms.base.Transform attribute) Transform (class in omniloader.transforms.base) TRANSFORMS (in module omniloader.transforms)

U

unified_collate() (in module omniloader.collate)

UnifiedSchema (class in omniloader.schema.spec) union_keys (omniloader.introspection.Report attribute)

V

val_dataloader() (omniloader.integrations.lightning.OmniDataModule method) valid_fraction (omniloader.introspection.Report attribute)

validate() (in module omniloader.introspection) (omniloader.loader.OmniLoader method) value_shape (omniloader.schema.spec.TensorSpec property)