torch_geometric.datasets.MovieLens

class MovieLens(root: str, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, model_name: Optional[str] = 'all-MiniLM-L6-v2', force_reload: bool = False)[source]

Bases: InMemoryDataset

一个异构的评分数据集，由GroupLens Research从MovieLens网站收集，包含类型为"movie"和"user"的节点。 用户对电影的评分可作为用户与电影之间边的真实标签("user", "rates", "movie")。

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)

  • model_name (str) – 用于将电影标题转换为节点特征的模型名称。该模型来自`Huggingface SentenceTransformer <https://huggingface.co/sentence-transformers>`_.

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