mmlearn.datasets.core package

Modules for core dataloading functionality.

Submodules

• mmlearn.datasets.core.combined_dataset module
  • CombinedDataset
    • CombinedDataset.__getitem__()
    • CombinedDataset.__len__()
• mmlearn.datasets.core.data_collator module
  • DefaultDataCollator
    • DefaultDataCollator.__call__()
    • DefaultDataCollator.batch_processors
  • collate_example_list()
• mmlearn.datasets.core.example module
  • Example
    • Example.create_ids()
  • find_matching_indices()
• mmlearn.datasets.core.modalities module
  • Modality
    • Modality.add_property()
    • Modality.attention_mask
    • Modality.ema_embedding
    • Modality.embedding
    • Modality.mask
    • Modality.masked_embedding
    • Modality.modality_specific_properties
    • Modality.name
    • Modality.properties
    • Modality.target
  • ModalityRegistry
    • ModalityRegistry.add_default_property()
    • ModalityRegistry.get_modality()
    • ModalityRegistry.get_modality_properties()
    • ModalityRegistry.has_modality()
    • ModalityRegistry.list_modalities()
    • ModalityRegistry.register_modality()
• mmlearn.datasets.core.samplers module
  • CombinedDatasetRatioSampler
    • CombinedDatasetRatioSampler.dataset
    • CombinedDatasetRatioSampler.num_samples
    • CombinedDatasetRatioSampler.probs
    • CombinedDatasetRatioSampler.replacement
    • CombinedDatasetRatioSampler.shuffle
    • CombinedDatasetRatioSampler.rank
    • CombinedDatasetRatioSampler.num_replicas
    • CombinedDatasetRatioSampler.drop_last
    • CombinedDatasetRatioSampler.seed
    • CombinedDatasetRatioSampler.epoch
    • CombinedDatasetRatioSampler.total_size
    • CombinedDatasetRatioSampler.__iter__()
    • CombinedDatasetRatioSampler.__len__()
    • CombinedDatasetRatioSampler.num_samples
    • CombinedDatasetRatioSampler.set_epoch()
    • CombinedDatasetRatioSampler.total_size
  • DistributedEvalSampler
    • DistributedEvalSampler.__iter__()
    • DistributedEvalSampler.__len__()
    • DistributedEvalSampler.num_samples
    • DistributedEvalSampler.set_epoch()
    • DistributedEvalSampler.total_size