topobench.data.loaders.graph package

Init file for graph load module with automated loader discovery.

class ADMEDatasetLoader(parameters)

Bases: AbstractLoader

Load TDC ADME datasets with SMILES to graph conversion using OGB featurization.

This loader: 1. Loads ADME datasets from TDC (Therapeutics Data Commons) 2. Converts SMILES strings to PyG graphs using OGB’s standard featurization 3. Uses fixed scaffold splits from TDC 4. Returns graphs compatible with OGB molecular property prediction

Node features (9-dimensional):

• Atomic number

• Chirality

• Degree

• Formal charge

• Number of hydrogens

• Number of radical electrons

• Hybridization

• Is aromatic

• Is in ring

Edge features (3-dimensional):

• Bond type

• Bond stereochemistry

• Is conjugated

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the ADME dataset

• data_type: Type of the dataset (e.g., “ADME”)

__init__(parameters)

get_data_dir()

Get the data directory.

Returns:

Path

The path to the dataset directory. Format: {root_data_dir}/{dataset_name}/. Example: data/graph/ADME/BBB_Martins/.

load_dataset()

Load the ADME dataset with predefined scaffold splits.

Returns:

InMemoryDataset

The dataset with converted graphs and predefined splits.

Raises:

RuntimeError

If dataset loading or SMILES conversion fails.

ValueError

If invalid SMILES strings are encountered.

ImportError

If PyTDC or rdkit (via ogb) are not installed.

class GraphUniverseDatasetLoader(parameters)

Bases: AbstractLoader

Load Graph Universe datasets.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• data_type: Type of the dataset (e.g., “graph_classification”)

__init__(parameters)

load(**kwargs)

Load data.

Parameters:

**kwargsdict

Additional keyword arguments.

Returns:

tuple[torch_geometric.data.Data, str]

Tuple containing the loaded data and the data directory.

load_dataset()

Load Graph Universe dataset.

Returns:

Dataset

The loaded Graph Universe dataset.

Raises:

RuntimeError

If dataset loading fails.

class HeterophilousGraphDatasetLoader(parameters)

Bases: AbstractLoader

Load Heterophilous Graph datasets.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• data_type: Type of the dataset (e.g., “heterophilous”)

__init__(parameters)

load_dataset()

Load Heterophilous Graph dataset.

Returns:

Dataset

The loaded Heterophilous Graph dataset.

Raises:

RuntimeError

If dataset loading fails.

class MantraSimplicialDatasetLoader(parameters, **kwargs)

Bases: AbstractLoader

Load Mantra dataset with configurable parameters.

Note: for the simplicial datasets it is necessary to include DatasetLoader into the name of the class!

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• other relevant parameters

**kwargsdict

Additional keyword arguments.

__init__(parameters, **kwargs)

load_dataset(**kwargs)

Load the MANTRA dataset.

Parameters:

**kwargsdict

Additional keyword arguments for dataset initialization.

Returns:

CitationHypergraphDataset

The loaded Citation Hypergraph dataset with the appropriate data_dir.

Raises:

RuntimeError

If dataset loading fails.

class ManualGraphDatasetLoader(parameters)

Bases: AbstractLoader

Load manually provided graph datasets.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_name: Name of the dataset

• data_dir: Root directory for data

__init__(parameters)

get_data_dir()

Get the data directory.

Returns:

Path

The path to the dataset directory.

load_dataset()

Load the manual graph dataset.

Returns:

DataloadDataset

The dataset object containing the manually loaded graph.

class MoleculeDatasetLoader(parameters)

Bases: AbstractLoader

Load molecule datasets (ZINC and AQSOL) with predefined splits, or QM9.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• data_type: Type of the dataset (e.g., “molecule”)

• qm9_target_index: (QM9 only) Which of the 19 regression targets to use (default 0).

__init__(parameters)

get_data_dir()

Get the data directory.

Returns:

Path

The path to the dataset directory.

load_dataset()

Load the molecule dataset with predefined splits.

Returns:

Dataset

The combined dataset with predefined splits.

Raises:

RuntimeError

If dataset loading fails.

class OGBGDatasetLoader(parameters)

Bases: AbstractLoader

Load molecule datasets (molhiv, molpcba, ppa) with predefined splits.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• data_type: Type of the dataset (e.g., “molecule”)

__init__(parameters)

get_data_dir()

Get the data directory.

Returns:

Path

The path to the dataset directory.

load_dataset()

Load the molecule dataset with predefined splits.

Returns:

Dataset

The combined dataset with predefined splits.

Raises:

RuntimeError

If dataset loading fails.

class PlanetoidDatasetLoader(parameters)

Bases: AbstractLoader

Load PLANETOID datasets.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• data_type: Type of the dataset (e.g., “cocitation”)

__init__(parameters)

load_dataset()

Load Planetoid dataset.

Returns:

Dataset

The loaded Planetoid dataset.

Raises:

RuntimeError

If dataset loading fails.

class TUDatasetLoader(parameters)

Bases: AbstractLoader

Load TU datasets.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• data_type: Type of the dataset (e.g., “graph_classification”)

__init__(parameters)

load_dataset()

Load TU dataset.

Returns:

Dataset

The loaded TU dataset.

Raises:

RuntimeError

If dataset loading fails.

class USCountyDemosDatasetLoader(parameters)

Bases: AbstractLoader

Load US County Demos dataset with configurable year and task variable.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• year: Year of the dataset (if applicable)

• task_variable: Task variable for the dataset

__init__(parameters)

load_dataset()

Load the US County Demos dataset.

Returns:

USCountyDemosDataset

The loaded US County Demos dataset with the appropriate data_dir.

Raises:

RuntimeError

If dataset loading fails.

Submodules

• topobench.data.loaders.graph.adme_datasets module
  • ADME
    • ADME.__init__()
    • ADME.get_approved_set()
    • ADME.get_other_species()
    • ADME.harmonize()
  • ADMEDatasetLoader
    • ADMEDatasetLoader.__init__()
    • ADMEDatasetLoader.get_data_dir()
    • ADMEDatasetLoader.load_dataset()
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Data
    • Data.from_dict()
    • Data.__init__()
    • Data.connected_components()
    • Data.debug()
    • Data.edge_subgraph()
    • Data.get_all_edge_attrs()
    • Data.get_all_tensor_attrs()
    • Data.is_edge_attr()
    • Data.is_node_attr()
    • Data.stores_as()
    • Data.subgraph()
    • Data.to_dict()
    • Data.to_heterogeneous()
    • Data.to_namedtuple()
    • Data.update()
    • Data.validate()
    • Data.batch
    • Data.edge_attr
    • Data.edge_index
    • Data.edge_stores
    • Data.edge_weight
    • Data.face
    • Data.node_stores
    • Data.num_edge_features
    • Data.num_edge_types
    • Data.num_faces
    • Data.num_features
    • Data.num_node_features
    • Data.num_node_types
    • Data.num_nodes
    • Data.pos
    • Data.stores
    • Data.time
    • Data.x
    • Data.y
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • InMemoryDataset
    • InMemoryDataset.save()
    • InMemoryDataset.collate()
    • InMemoryDataset.__init__()
    • InMemoryDataset.copy()
    • InMemoryDataset.cpu()
    • InMemoryDataset.cuda()
    • InMemoryDataset.get()
    • InMemoryDataset.len()
    • InMemoryDataset.load()
    • InMemoryDataset.to()
    • InMemoryDataset.to_on_disk_dataset()
    • InMemoryDataset.data
    • InMemoryDataset.num_classes
    • InMemoryDataset.processed_file_names
    • InMemoryDataset.raw_file_names
  • Path
    • Path.cwd()
    • Path.home()
    • Path.absolute()
    • Path.chmod()
    • Path.exists()
    • Path.expanduser()
    • Path.glob()
    • Path.group()
    • Path.hardlink_to()
    • Path.is_block_device()
    • Path.is_char_device()
    • Path.is_dir()
    • Path.is_fifo()
    • Path.is_file()
    • Path.is_mount()
    • Path.is_socket()
    • Path.is_symlink()
    • Path.iterdir()
    • Path.lchmod()
    • Path.link_to()
    • Path.lstat()
    • Path.mkdir()
    • Path.open()
    • Path.owner()
    • Path.read_bytes()
    • Path.read_text()
    • Path.readlink()
    • Path.rename()
    • Path.replace()
    • Path.resolve()
    • Path.rglob()
    • Path.rmdir()
    • Path.samefile()
    • Path.stat()
    • Path.symlink_to()
    • Path.touch()
    • Path.unlink()
    • Path.write_bytes()
    • Path.write_text()
  • smiles2graph()
• topobench.data.loaders.graph.graph_universe_loader module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Data
    • Data.from_dict()
    • Data.__init__()
    • Data.connected_components()
    • Data.debug()
    • Data.edge_subgraph()
    • Data.get_all_edge_attrs()
    • Data.get_all_tensor_attrs()
    • Data.is_edge_attr()
    • Data.is_node_attr()
    • Data.stores_as()
    • Data.subgraph()
    • Data.to_dict()
    • Data.to_heterogeneous()
    • Data.to_namedtuple()
    • Data.update()
    • Data.validate()
    • Data.batch
    • Data.edge_attr
    • Data.edge_index
    • Data.edge_stores
    • Data.edge_weight
    • Data.face
    • Data.node_stores
    • Data.num_edge_features
    • Data.num_edge_types
    • Data.num_faces
    • Data.num_features
    • Data.num_node_features
    • Data.num_node_types
    • Data.num_nodes
    • Data.pos
    • Data.stores
    • Data.time
    • Data.x
    • Data.y
  • Dataset
    • Dataset.__init__()
    • Dataset.download()
    • Dataset.get()
    • Dataset.get_summary()
    • Dataset.index_select()
    • Dataset.indices()
    • Dataset.len()
    • Dataset.print_summary()
    • Dataset.process()
    • Dataset.shuffle()
    • Dataset.to_datapipe()
    • Dataset.has_download
    • Dataset.has_process
    • Dataset.num_classes
    • Dataset.num_edge_features
    • Dataset.num_features
    • Dataset.num_node_features
    • Dataset.processed_dir
    • Dataset.processed_file_names
    • Dataset.processed_paths
    • Dataset.raw_dir
    • Dataset.raw_file_names
    • Dataset.raw_paths
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • GraphUniverseDataset
    • GraphUniverseDataset.__init__()
    • GraphUniverseDataset.download()
    • GraphUniverseDataset.get_data_dir()
    • GraphUniverseDataset.get_dataset_dir()
    • GraphUniverseDataset.process()
    • GraphUniverseDataset.processed_dir
    • GraphUniverseDataset.processed_file_names
    • GraphUniverseDataset.raw_dir
    • GraphUniverseDataset.raw_file_names
  • GraphUniverseDatasetLoader
    • GraphUniverseDatasetLoader.__init__()
    • GraphUniverseDatasetLoader.load()
    • GraphUniverseDatasetLoader.load_dataset()
• topobench.data.loaders.graph.hetero_datasets module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Dataset
    • Dataset.__init__()
    • Dataset.download()
    • Dataset.get()
    • Dataset.get_summary()
    • Dataset.index_select()
    • Dataset.indices()
    • Dataset.len()
    • Dataset.print_summary()
    • Dataset.process()
    • Dataset.shuffle()
    • Dataset.to_datapipe()
    • Dataset.has_download
    • Dataset.has_process
    • Dataset.num_classes
    • Dataset.num_edge_features
    • Dataset.num_features
    • Dataset.num_node_features
    • Dataset.processed_dir
    • Dataset.processed_file_names
    • Dataset.processed_paths
    • Dataset.raw_dir
    • Dataset.raw_file_names
    • Dataset.raw_paths
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • HeterophilousGraphDataset
    • HeterophilousGraphDataset.__init__()
    • HeterophilousGraphDataset.download()
    • HeterophilousGraphDataset.process()
    • HeterophilousGraphDataset.processed_dir
    • HeterophilousGraphDataset.processed_file_names
    • HeterophilousGraphDataset.raw_dir
    • HeterophilousGraphDataset.raw_file_names
    • HeterophilousGraphDataset.url
  • HeterophilousGraphDatasetLoader
    • HeterophilousGraphDatasetLoader.__init__()
    • HeterophilousGraphDatasetLoader.load_dataset()
• topobench.data.loaders.graph.mantra_dataset module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • MantraDataset
    • MantraDataset.__init__()
    • MantraDataset.download()
    • MantraDataset.process()
    • MantraDataset.FILE_FORMAT
    • MantraDataset.RAW_FILE_NAMES
    • MantraDataset.URLS
    • MantraDataset.processed_dir
    • MantraDataset.processed_file_names
    • MantraDataset.raw_dir
    • MantraDataset.raw_file_names
    • MantraDataset.slices
  • MantraSimplicialDatasetLoader
    • MantraSimplicialDatasetLoader.__init__()
    • MantraSimplicialDatasetLoader.load_dataset()
• topobench.data.loaders.graph.manual_graph_dataset_loader module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Any
  • DataloadDataset
    • DataloadDataset.__init__()
    • DataloadDataset.get()
    • DataloadDataset.len()
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • ManualGraphDatasetLoader
    • ManualGraphDatasetLoader.__init__()
    • ManualGraphDatasetLoader.get_data_dir()
    • ManualGraphDatasetLoader.load_dataset()
  • Path
    • Path.cwd()
    • Path.home()
    • Path.absolute()
    • Path.chmod()
    • Path.exists()
    • Path.expanduser()
    • Path.glob()
    • Path.group()
    • Path.hardlink_to()
    • Path.is_block_device()
    • Path.is_char_device()
    • Path.is_dir()
    • Path.is_fifo()
    • Path.is_file()
    • Path.is_mount()
    • Path.is_socket()
    • Path.is_symlink()
    • Path.iterdir()
    • Path.lchmod()
    • Path.link_to()
    • Path.lstat()
    • Path.mkdir()
    • Path.open()
    • Path.owner()
    • Path.read_bytes()
    • Path.read_text()
    • Path.readlink()
    • Path.rename()
    • Path.replace()
    • Path.resolve()
    • Path.rglob()
    • Path.rmdir()
    • Path.samefile()
    • Path.stat()
    • Path.symlink_to()
    • Path.touch()
    • Path.unlink()
    • Path.write_bytes()
    • Path.write_text()
  • load_manual_graph()
• topobench.data.loaders.graph.molecule_datasets module
  • AQSOL
    • AQSOL.__init__()
    • AQSOL.atoms()
    • AQSOL.bonds()
    • AQSOL.download()
    • AQSOL.process()
    • AQSOL.processed_file_names
    • AQSOL.raw_file_names
    • AQSOL.url
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Dataset
    • Dataset.__init__()
    • Dataset.download()
    • Dataset.get()
    • Dataset.get_summary()
    • Dataset.index_select()
    • Dataset.indices()
    • Dataset.len()
    • Dataset.print_summary()
    • Dataset.process()
    • Dataset.shuffle()
    • Dataset.to_datapipe()
    • Dataset.has_download
    • Dataset.has_process
    • Dataset.num_classes
    • Dataset.num_edge_features
    • Dataset.num_features
    • Dataset.num_node_features
    • Dataset.processed_dir
    • Dataset.processed_file_names
    • Dataset.processed_paths
    • Dataset.raw_dir
    • Dataset.raw_file_names
    • Dataset.raw_paths
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • MoleculeDatasetLoader
    • MoleculeDatasetLoader.__init__()
    • MoleculeDatasetLoader.get_data_dir()
    • MoleculeDatasetLoader.load_dataset()
  • OmegaConf
    • OmegaConf.clear_resolver()
    • OmegaConf.has_resolver()
    • OmegaConf.clear_cache()
    • OmegaConf.clear_resolvers()
    • OmegaConf.copy_cache()
    • OmegaConf.create()
    • OmegaConf.from_cli()
    • OmegaConf.from_dotlist()
    • OmegaConf.get_cache()
    • OmegaConf.get_type()
    • OmegaConf.is_config()
    • OmegaConf.is_dict()
    • OmegaConf.is_interpolation()
    • OmegaConf.is_list()
    • OmegaConf.is_missing()
    • OmegaConf.is_readonly()
    • OmegaConf.is_struct()
    • OmegaConf.legacy_register_resolver()
    • OmegaConf.load()
    • OmegaConf.masked_copy()
    • OmegaConf.merge()
    • OmegaConf.missing_keys()
    • OmegaConf.register_new_resolver()
    • OmegaConf.register_resolver()
    • OmegaConf.resolve()
    • OmegaConf.save()
    • OmegaConf.select()
    • OmegaConf.set_cache()
    • OmegaConf.set_readonly()
    • OmegaConf.set_struct()
    • OmegaConf.structured()
    • OmegaConf.to_container()
    • OmegaConf.to_object()
    • OmegaConf.to_yaml()
    • OmegaConf.unsafe_merge()
    • OmegaConf.update()
    • OmegaConf.__init__()
  • Path
    • Path.cwd()
    • Path.home()
    • Path.absolute()
    • Path.chmod()
    • Path.exists()
    • Path.expanduser()
    • Path.glob()
    • Path.group()
    • Path.hardlink_to()
    • Path.is_block_device()
    • Path.is_char_device()
    • Path.is_dir()
    • Path.is_fifo()
    • Path.is_file()
    • Path.is_mount()
    • Path.is_socket()
    • Path.is_symlink()
    • Path.iterdir()
    • Path.lchmod()
    • Path.link_to()
    • Path.lstat()
    • Path.mkdir()
    • Path.open()
    • Path.owner()
    • Path.read_bytes()
    • Path.read_text()
    • Path.readlink()
    • Path.rename()
    • Path.replace()
    • Path.resolve()
    • Path.rglob()
    • Path.rmdir()
    • Path.samefile()
    • Path.stat()
    • Path.symlink_to()
    • Path.touch()
    • Path.unlink()
    • Path.write_bytes()
    • Path.write_text()
  • QM9
    • QM9.__init__()
    • QM9.atomref()
    • QM9.download()
    • QM9.mean()
    • QM9.process()
    • QM9.std()
    • QM9.processed_file_names
    • QM9.processed_url
    • QM9.raw_file_names
    • QM9.raw_url
    • QM9.raw_url2
  • ZINC
    • ZINC.__init__()
    • ZINC.download()
    • ZINC.process()
    • ZINC.processed_dir
    • ZINC.processed_file_names
    • ZINC.raw_file_names
    • ZINC.split_url
    • ZINC.url
• topobench.data.loaders.graph.ogbg_datasets module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Dataset
    • Dataset.__init__()
    • Dataset.download()
    • Dataset.get()
    • Dataset.get_summary()
    • Dataset.index_select()
    • Dataset.indices()
    • Dataset.len()
    • Dataset.print_summary()
    • Dataset.process()
    • Dataset.shuffle()
    • Dataset.to_datapipe()
    • Dataset.has_download
    • Dataset.has_process
    • Dataset.num_classes
    • Dataset.num_edge_features
    • Dataset.num_features
    • Dataset.num_node_features
    • Dataset.processed_dir
    • Dataset.processed_file_names
    • Dataset.processed_paths
    • Dataset.raw_dir
    • Dataset.raw_file_names
    • Dataset.raw_paths
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • OGBGDatasetLoader
    • OGBGDatasetLoader.__init__()
    • OGBGDatasetLoader.get_data_dir()
    • OGBGDatasetLoader.load_dataset()
  • Path
    • Path.cwd()
    • Path.home()
    • Path.absolute()
    • Path.chmod()
    • Path.exists()
    • Path.expanduser()
    • Path.glob()
    • Path.group()
    • Path.hardlink_to()
    • Path.is_block_device()
    • Path.is_char_device()
    • Path.is_dir()
    • Path.is_fifo()
    • Path.is_file()
    • Path.is_mount()
    • Path.is_socket()
    • Path.is_symlink()
    • Path.iterdir()
    • Path.lchmod()
    • Path.link_to()
    • Path.lstat()
    • Path.mkdir()
    • Path.open()
    • Path.owner()
    • Path.read_bytes()
    • Path.read_text()
    • Path.readlink()
    • Path.rename()
    • Path.replace()
    • Path.resolve()
    • Path.rglob()
    • Path.rmdir()
    • Path.samefile()
    • Path.stat()
    • Path.symlink_to()
    • Path.touch()
    • Path.unlink()
    • Path.write_bytes()
    • Path.write_text()
  • PygGraphPropPredDataset
    • PygGraphPropPredDataset.__init__()
    • PygGraphPropPredDataset.download()
    • PygGraphPropPredDataset.get_idx_split()
    • PygGraphPropPredDataset.process()
    • PygGraphPropPredDataset.num_classes
    • PygGraphPropPredDataset.processed_file_names
    • PygGraphPropPredDataset.raw_file_names
• topobench.data.loaders.graph.planetoid_datasets module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Dataset
    • Dataset.__init__()
    • Dataset.download()
    • Dataset.get()
    • Dataset.get_summary()
    • Dataset.index_select()
    • Dataset.indices()
    • Dataset.len()
    • Dataset.print_summary()
    • Dataset.process()
    • Dataset.shuffle()
    • Dataset.to_datapipe()
    • Dataset.has_download
    • Dataset.has_process
    • Dataset.num_classes
    • Dataset.num_edge_features
    • Dataset.num_features
    • Dataset.num_node_features
    • Dataset.processed_dir
    • Dataset.processed_file_names
    • Dataset.processed_paths
    • Dataset.raw_dir
    • Dataset.raw_file_names
    • Dataset.raw_paths
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • Planetoid
    • Planetoid.__init__()
    • Planetoid.download()
    • Planetoid.process()
    • Planetoid.geom_gcn_url
    • Planetoid.processed_dir
    • Planetoid.processed_file_names
    • Planetoid.raw_dir
    • Planetoid.raw_file_names
    • Planetoid.url
  • PlanetoidDatasetLoader
    • PlanetoidDatasetLoader.__init__()
    • PlanetoidDatasetLoader.load_dataset()
• topobench.data.loaders.graph.tu_datasets module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • Dataset
    • Dataset.__init__()
    • Dataset.download()
    • Dataset.get()
    • Dataset.get_summary()
    • Dataset.index_select()
    • Dataset.indices()
    • Dataset.len()
    • Dataset.print_summary()
    • Dataset.process()
    • Dataset.shuffle()
    • Dataset.to_datapipe()
    • Dataset.has_download
    • Dataset.has_process
    • Dataset.num_classes
    • Dataset.num_edge_features
    • Dataset.num_features
    • Dataset.num_node_features
    • Dataset.processed_dir
    • Dataset.processed_file_names
    • Dataset.processed_paths
    • Dataset.raw_dir
    • Dataset.raw_file_names
    • Dataset.raw_paths
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • TUDataset
    • TUDataset.__init__()
    • TUDataset.download()
    • TUDataset.process()
    • TUDataset.cleaned_url
    • TUDataset.num_edge_attributes
    • TUDataset.num_edge_labels
    • TUDataset.num_node_attributes
    • TUDataset.num_node_labels
    • TUDataset.processed_dir
    • TUDataset.processed_file_names
    • TUDataset.raw_dir
    • TUDataset.raw_file_names
    • TUDataset.url
  • TUDatasetLoader
    • TUDatasetLoader.__init__()
    • TUDatasetLoader.load_dataset()
• topobench.data.loaders.graph.us_county_demos_dataset_loader module
  • AbstractLoader
    • AbstractLoader.__init__()
    • AbstractLoader.get_data_dir()
    • AbstractLoader.load()
    • AbstractLoader.load_dataset()
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.items()
    • DictConfig.items_ex()
    • DictConfig.keys()
    • DictConfig.pop()
    • DictConfig.setdefault()
  • Path
    • Path.cwd()
    • Path.home()
    • Path.absolute()
    • Path.chmod()
    • Path.exists()
    • Path.expanduser()
    • Path.glob()
    • Path.group()
    • Path.hardlink_to()
    • Path.is_block_device()
    • Path.is_char_device()
    • Path.is_dir()
    • Path.is_fifo()
    • Path.is_file()
    • Path.is_mount()
    • Path.is_socket()
    • Path.is_symlink()
    • Path.iterdir()
    • Path.lchmod()
    • Path.link_to()
    • Path.lstat()
    • Path.mkdir()
    • Path.open()
    • Path.owner()
    • Path.read_bytes()
    • Path.read_text()
    • Path.readlink()
    • Path.rename()
    • Path.replace()
    • Path.resolve()
    • Path.rglob()
    • Path.rmdir()
    • Path.samefile()
    • Path.stat()
    • Path.symlink_to()
    • Path.touch()
    • Path.unlink()
    • Path.write_bytes()
    • Path.write_text()
  • USCountyDemosDataset
    • USCountyDemosDataset.__init__()
    • USCountyDemosDataset.download()
    • USCountyDemosDataset.process()
    • USCountyDemosDataset.FILE_FORMAT
    • USCountyDemosDataset.RAW_FILE_NAMES
    • USCountyDemosDataset.URLS
    • USCountyDemosDataset.processed_dir
    • USCountyDemosDataset.processed_file_names
    • USCountyDemosDataset.raw_dir
    • USCountyDemosDataset.raw_file_names
    • USCountyDemosDataset.slices
  • USCountyDemosDatasetLoader
    • USCountyDemosDatasetLoader.__init__()
    • USCountyDemosDatasetLoader.load_dataset()