torch_geometric.datasets.LastFM

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

Bases: InMemoryDataset

last.fm音乐网站的一个子集，用于跟踪用户从各种来源的收听信息，这些信息收集在“MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding”论文中。last.fm是一个包含三种类型实体的异构图——用户（1,892个节点）、艺术家（17,632个节点）和艺术家标签（1,088个节点）。该数据集可用于链接预测，且不提供标签或特征。

Parameters:

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

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