API#

AnnData to Data#

Datamodule#

class geome.datamodule.GraphAnnDataModule(datas=None, batch_size=1, num_workers=1, learning_type='node')#

Lightning DataModule for graph data.

setup(stage=None)#: Setup function to be called at the beginning of training, validation or testing.

Args:#

stage (str, optional): the stage of the training, either ‘train’, ‘val’ or ‘test’. Defaults to None.

test_dataloader()#: Returns the test dataloader.

train_dataloader()#: Returns the training dataloader.

val_dataloader()#: Returns the validation dataloader.