roksana.datasets.datasets

Functions

get_dataset_info(dataset)

Retrieve basic information about a dataset.

list_available_standard_datasets()

List all available standard datasets supported by ROKSANA.

load_dataset([dataset_name, root, ...])

Load a dataset, either a standard dataset or a user-provided dataset.

load_standard_dataset(name[, root])

Load a standard dataset from PyG's built-in datasets.

load_user_dataset_from_files(data_dir[, ...])

Load a user dataset from files in a specified directory.

prepare_search_set(data[, percentage, seed])

Prepare a search set for search evaluation by selecting a percentage of nodes as queries and creating corresponding gold sets based on feature similarity.

Classes

UserDataset(root[, transform, ...])

A dataset class for user-provided datasets adhering to PyG's InMemoryDataset structure.