quaterion.dataset package

Submodules

• quaterion.dataset.indexing_dataset module
  • IndexingDataset
    • IndexingDataset.set_salt()
    • IndexingDataset.set_skip_read()
  • IndexingIterableDataset
    • IndexingIterableDataset.set_salt()
    • IndexingIterableDataset.set_skip_read()
• quaterion.dataset.label_cache_dataset module
  • LabelCacheDataset
  • LabelCacheDatasetMixin
    • LabelCacheDatasetMixin.load()
    • LabelCacheDatasetMixin.process_item()
    • LabelCacheDatasetMixin.save()
    • LabelCacheDatasetMixin.set_mode()
    • LabelCacheDatasetMixin.mode
  • LabelCacheIterableDataset
  • LabelCacheMode
    • LabelCacheMode.learn
    • LabelCacheMode.read
    • LabelCacheMode.transparent
• quaterion.dataset.similarity_data_loader module
  • GroupSimilarityDataLoader
    • GroupSimilarityDataLoader.collate_labels()
    • GroupSimilarityDataLoader.flatten_objects()
    • GroupSimilarityDataLoader.batch_size
    • GroupSimilarityDataLoader.dataset
    • GroupSimilarityDataLoader.drop_last
    • GroupSimilarityDataLoader.num_workers
    • GroupSimilarityDataLoader.pin_memory
    • GroupSimilarityDataLoader.pin_memory_device
    • GroupSimilarityDataLoader.prefetch_factor
    • GroupSimilarityDataLoader.sampler
    • GroupSimilarityDataLoader.timeout
  • PairsSimilarityDataLoader
    • PairsSimilarityDataLoader.collate_labels()
    • PairsSimilarityDataLoader.flatten_objects()
    • PairsSimilarityDataLoader.batch_size
    • PairsSimilarityDataLoader.dataset
    • PairsSimilarityDataLoader.drop_last
    • PairsSimilarityDataLoader.num_workers
    • PairsSimilarityDataLoader.pin_memory
    • PairsSimilarityDataLoader.pin_memory_device
    • PairsSimilarityDataLoader.prefetch_factor
    • PairsSimilarityDataLoader.sampler
    • PairsSimilarityDataLoader.timeout
  • SimilarityDataLoader
    • SimilarityDataLoader.collate_labels()
    • SimilarityDataLoader.flatten_objects()
    • SimilarityDataLoader.load_label_cache()
    • SimilarityDataLoader.pre_collate_fn()
    • SimilarityDataLoader.save_label_cache()
    • SimilarityDataLoader.set_label_cache_mode()
    • SimilarityDataLoader.set_salt()
    • SimilarityDataLoader.set_skip_read()
    • SimilarityDataLoader.batch_size
    • SimilarityDataLoader.dataset
    • SimilarityDataLoader.drop_last
    • SimilarityDataLoader.full_cache_used
    • SimilarityDataLoader.num_workers
    • SimilarityDataLoader.original_params
    • SimilarityDataLoader.pin_memory
    • SimilarityDataLoader.pin_memory_device
    • SimilarityDataLoader.prefetch_factor
    • SimilarityDataLoader.sampler
    • SimilarityDataLoader.timeout
• quaterion.dataset.similarity_dataset module
  • SimilarityGroupDataset
• quaterion.dataset.similarity_samples module
  • SimilarityGroupSample
    • SimilarityGroupSample.group
    • SimilarityGroupSample.obj
  • SimilarityPairSample
    • SimilarityPairSample.obj_a
    • SimilarityPairSample.obj_b
    • SimilarityPairSample.score
    • SimilarityPairSample.subgroup
• quaterion.dataset.train_collator module
  • TrainCollator
    • TrainCollator.pre_encoder_collate()
    • TrainCollator.process_meta()

Module contents