torch_geometric.datasets.IGMCDataset

class IGMCDataset(root: str, name: str, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

用户-项目异构评分数据集 "Douban", "Flixster" 和 "Yahoo-Music" 来自 “基于图神经网络的归纳矩阵完成” 论文。

节点代表用户和物品。 用户和物品之间的边和特征代表用户对物品的（训练）评分。

Parameters:

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

  • name (str) – 数据集的名称 ("Douban", "Flixster", "Yahoo-Music").

  • transform (callable, optional) – A function/transform that takes in an torch_geometric.data.HeteroData 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.HeteroData object and returns a transformed version. The data object will be transformed before being saved to disk. (default: None)

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