torch_adata._tools._base_lightning_data_module

Module Contents

Classes

BaseLightningDataModule

Helper class that provides a standard way to create an ABC using

Attributes

__module_name__

__doc__

__author__

__email__

torch_adata._tools._base_lightning_data_module.__module_name__ = _base_lightning_data_module.py
torch_adata._tools._base_lightning_data_module.__doc__ = Aux. module to organize AnnData/torch datasets into PyTorch-Lightning LightningDataModule.
torch_adata._tools._base_lightning_data_module.__author__
torch_adata._tools._base_lightning_data_module.__email__
class torch_adata._tools._base_lightning_data_module.BaseLightningDataModule(adata: anndata.AnnData = None, batch_size: int = 2000, num_workers: int = os.cpu_count(), **kwargs)

Bases: abc.ABC, pytorch_lightning.LightningDataModule

Helper class that provides a standard way to create an ABC using inheritance.

__parse__(kwargs, ignore=['self', '__class__'])
train_dataloader()
val_dataloader()
test_dataloader()
predict_dataloader()