topobench.data.loaders.simplicial package#

Submodules#

topobench.data.loaders.simplicial.mantra_dataset_loader module#

Loaders for Mantra dataset as simplicial.

class topobench.data.loaders.simplicial.mantra_dataset_loader.MantraSimplicialDatasetLoader(parameters: DictConfig, **kwargs)[source]#

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.

load_dataset(**kwargs) → MantraDataset[source]#

Load the Citation Hypergraph 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.

Module contents#

Init file for simplicial dataset load module with automated loader discovery.

class topobench.data.loaders.simplicial.MantraSimplicialDatasetLoader(parameters: DictConfig, **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.

load_dataset(**kwargs) → MantraDataset#

Load the Citation Hypergraph 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.