torch_geometric.datasets.CornellTemporalHyperGraphDataset

class CornellTemporalHyperGraphDataset(root: str, name: str, split: str = 'train', setting: str = 'transductive', transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, pre_filter: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

来自“Simplicial Closure and higher-order link prediction”论文的时间高阶网络数据集集合。 每个数据集都是一个带时间戳的单纯形序列，其中单纯形是一组\(k\)个节点。

有关各个数据集的更多详细信息，请参阅原始数据集页面。

Parameters:

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

  • name (str) – The name of the dataset.

  • split (str, optional) – If "train", loads the training dataset. If "val", loads the validation dataset. If "test", loads the test dataset. (default: "train")

  • 设置 (str, 可选) – 如果为 "transductive"，则加载数据集用于传导训练。 如果为 "inductive"，则加载数据集用于归纳训练。 (默认: "transductive")

  • 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)

  • pre_filter (callable, optional) – A function that takes in an torch_geometric.data.Data object and returns a boolean value, indicating whether the data object should be included in the final dataset. (default: None)

  • force_reload (bool, optional) – Whether to re-process the dataset. (default: False)