datasets

Submodules

• datasets.gatne
• datasets.gcc_data
• datasets.gtn_data
• datasets.han_data
• datasets.matlab_matrix
• datasets.pyg
• datasets.pyg_modelnet

Package Contents

Functions

register_dataset(name)	New dataset types can be added to cogdl with the register_dataset()
build_dataset(args)
build_dataset_from_name(dataset)

datasets.pyg = False

datasets.dgl_import = False

datasets.DATASET_REGISTRY

datasets.register_dataset(name)

New dataset types can be added to cogdl with the register_dataset() function decorator.

For example:

@register_dataset('my_dataset')
class MyDataset():
    (...)

Args:

name (str): the name of the dataset

datasets.dataset_name

datasets.build_dataset(args)

datasets.build_dataset_from_name(dataset)