torch_geometric_signed_directed.data.directed.DIGRAC_real_data

Classes

DIGRAC_real_data

Data loader for the data sets used in the

Module Contents

class DIGRAC_real_data(name: str, root: str, transform: Callable | None = None, pre_transform: Callable | None = None)

Bases: torch_geometric.data.InMemoryDataset

Data loader for the data sets used in the DIGRAC: Digraph Clustering Based on Flow Imbalance” paper.

Parameters:

  • name (str) – Name of the data set, choices are: ‘blog’, ‘wikitalk’, ‘migration’, ‘lead_lag”+str(year) (year from 2001 to 2019).

  • root (str) – Root directory where the dataset should be saved.

  • transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before every access. (default: None)

  • pre_transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before being saved to disk. (default: None)

name
url
property raw_file_names
property processed_file_names
download()
process()