GOOD.data.good_loaders.BaseLoader

Classes

BaseDataLoader(*args, **kwargs)

class GOOD.data.good_loaders.BaseLoader.BaseDataLoader(*args, **kwargs)[source]

Bases: Munch

classmethod setup(dataset, config: Union[CommonArgs, Munch])[source]

Create a PyG data loader.

Parameters

dataset – A GOOD dataset.
config – Required configs: config.train.train_bs config.train.val_bs config.train.test_bs config.model.model_layer config.train.num_steps(for node prediction)

Returns

A PyG dataset loader.