nebula.core.datasets.datamodule#
Attributes#
Classes#
Module Contents#
- nebula.core.datasets.datamodule.logging_training#
- class nebula.core.datasets.datamodule.DataModule(train_set, train_set_indices, test_set, test_set_indices, local_test_set_indices, partition_id=0, partitions_number=1, batch_size=32, num_workers=0, val_percent=0.1, label_flipping=False, data_poisoning=False, poisoned_persent=0, poisoned_ratio=0, targeted=False, target_label=0, target_changed_label=0, noise_type='salt')#
Bases:
lightning.LightningDataModule
- train_set#
- train_set_indices#
- test_set#
- test_set_indices#
- local_test_set_indices#
- partition_id#
- partitions_number#
- batch_size#
- num_workers#
- val_percent#
- label_flipping#
- data_poisoning#
- poisoned_percent#
- poisoned_ratio#
- targeted#
- target_label#
- target_changed_label#
- noise_type#
- tr_subset#
- train_size#
- val_size#
- global_te_subset#
- local_te_subset#
- train_loader#
- val_loader#
- test_loader#
- global_test_loader#
- random_sampler#
- bootstrap_loader#
- train_dataloader()#
- val_dataloader()#
- test_dataloader()#
- bootstrap_dataloader()#