Data Handling

Dataset and dataloader utilities.

• DataBatch - Data batches
• DatasetWeighted - Weighted datasets
• DataLoaderWeighted - Weighted dataloaders
• DataModuleWeighted - Lightning data module
• DataModuleWeightedKFold - Lightning data module with k-fold cross-validation

Memory-mapped datasets.

• DatasetWeightedMemoryMapped

Stratified dataloader.

• StratifiedSampler
• StratifiedBatchSampler

Data utilities.

• ConcatDatasetWeighted
• ChainDatasetWeighted

I/O.

• save_dataset
• load_dataset