torch_geometric_signed_directed.data.signed.SSSNET_real_data

Classes

SSSNET_real_data

Data loader for the data sets used in the

Module Contents

class SSSNET_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 SSSNET: Semi-Supervised Signed Network Clustering paper.

Parameters:

  • name (str) – Name of the data set, choices are: ‘rainfall’, ‘PPI’, ‘wikirfa’, ‘sampson’, ‘SP1500’, ‘Fin_YNet”+str(year) (year from 2000 to 2020).

  • 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 = 'https://github.com/SherylHYX/pytorch_geometric_signed_directed/raw/main/datasets/Sampson'
property raw_file_names
property processed_file_names
download()
process()