torch_geometric.datasets.Taobao

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

Bases: InMemoryDataset

淘宝是一个由阿里巴巴提供的用户行为数据集，由天池阿里云平台提供。

淘宝是一个用于推荐的异构图。 节点代表具有用户ID的用户、具有商品ID的商品和具有类别ID的类别。 用户和商品之间的边代表用户对商品的不同类型的行为（以及时间戳）。 商品和类别之间的边将每个商品分配到其类别集合中。

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)