topobench.data.loaders.hypergraph.citation_hypergraph_dataset_loader module

Loaders for Citation Hypergraph dataset.

class AbstractLoader(parameters)

Bases: ABC

Abstract class that provides an interface to load data.

Parameters:

parametersDictConfig

Configuration parameters.

__init__(parameters)

get_data_dir()

Get the data directory.

Returns:

Path

The path to the dataset directory.

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.

abstract load_dataset()

Load data into a dataset.

Returns:

Union[torch_geometric.data.Dataset, torch.utils.data.Dataset]

The loaded dataset, which could be a PyG or PyTorch dataset.

Raises:

NotImplementedError

If the method is not implemented.

class CitationHypergraphDataset(root, name, parameters)

Bases: InMemoryDataset

Dataset class for US County Demographics dataset.

Parameters:

rootstr

Root directory where the dataset will be saved.

namestr

Name of the dataset.

parametersDictConfig

Configuration parameters for the dataset.

Attributes:

URLS (dict): Dictionary containing the URLs for downloading the dataset.

FILE_FORMAT (dict): Dictionary containing the file formats for the dataset.

RAW_FILE_NAMES (dict): Dictionary containing the raw file names for the dataset.

__init__(root, name, parameters)

download()

Download the dataset from a URL and saves it to the raw directory.

Raises:

FileNotFoundError – If the dataset URL is not found.

process()

Handle the data for the dataset.

This method loads the US county demographics data, applies any pre- processing transformations if specified, and saves the processed data to the appropriate location.

FILE_FORMAT: ClassVar = {'coauthorship_cora': 'zip', 'coauthorship_dblp': 'zip', 'cocitation_citeseer': 'zip', 'cocitation_cora': 'zip', 'cocitation_pubmed': 'zip'}

RAW_FILE_NAMES: ClassVar = {}

URLS: ClassVar = {'coauthorship_cora': 'https://drive.google.com/file/d/1J5fLPABWrM9SH_7m85n7--oHDVmwJeib/view?usp=sharing', 'coauthorship_dblp': 'https://drive.google.com/file/d/16ryf4Ve-t0_nAla0VfjtSxSAG8Sye8TZ/view?usp=sharing', 'cocitation_citeseer': 'https://drive.google.com/file/d/1XWfu1jtijsmHmfCP6UQxyLsuPM8GBNJb/view?usp=sharing', 'cocitation_cora': 'https://drive.google.com/file/d/1WVRx5yDxSdZpvL6FK5Ji8H3lOnyYlraN/view?usp=sharing', 'cocitation_pubmed': 'https://drive.google.com/file/d/1XbqDJnHnV0HYvie3fcM8rquamnQsLTpK/view?usp=sharing'}

property processed_dir: str

Return the path to the processed directory of the dataset.

Returns:

str

Path to the processed directory.

property processed_file_names: str

Return the processed file name for the dataset.

Returns:

str

Processed file name.

property raw_dir: str

Return the path to the raw directory of the dataset.

Returns:

str

Path to the raw directory.

property raw_file_names: list[str]

Return the raw file names for the dataset.

Returns:

list[str]

List of raw file names.

slices: Dict[str, Tensor] | None

class CitationHypergraphDatasetLoader(parameters)

Bases: AbstractLoader

Load Citation Hypergraph dataset with configurable parameters.

Parameters:

parametersDictConfig

Configuration parameters containing:

• data_dir: Root directory for data

• data_name: Name of the dataset

• other relevant parameters

__init__(parameters)

load_dataset()

Load the Citation Hypergraph dataset.

Returns:

CitationHypergraphDataset

The loaded Citation Hypergraph dataset with the appropriate data_dir.

Raises:

RuntimeError

If dataset loading fails.

class DictConfig(content, key=None, parent=None, ref_type=typing.Any, key_type=typing.Any, element_type=typing.Any, is_optional=True, flags=None)

Bases: BaseContainer, MutableMapping[Any, Any]

__init__(content, key=None, parent=None, ref_type=typing.Any, key_type=typing.Any, element_type=typing.Any, is_optional=True, flags=None)

copy()

get(key, default_value=None)

Return the value for key if key is in the dictionary, else default_value (defaulting to None).

items() → a set-like object providing a view on D's items

items_ex(resolve=True, keys=None)

keys() → a set-like object providing a view on D's keys

pop(k[, d]) → v, remove specified key and return the corresponding value.

If key is not found, d is returned if given, otherwise KeyError is raised.

setdefault(k[, d]) → D.get(k,d), also set D[k]=d if k not in D