PyG构建电影评分MovieLens dataset图数据集_电影数据集

作者：盐析白兔 | 2024-06-24 11:16:15

踩

电影数据集

本文将使用 GroupLens 研究小组收集的 MovieLens数据集。这个数据集描述了来自MovieLens的5星级评级和标签活动，该数据集包含来自 600 多名用户的超过 9k 部电影的大约 10万次 评级。

我们将使用该数据集生成两种节点类型，分别保存电影和用户的数据，以及一种连接用户和电影的边缘类型，表示用户如何评价特定电影的关系。

在这里插入图片描述

下面我们将讲解如何基于 MovieLens dataset csv数据集转化为 Graph 图数据集：

1️⃣ 数据集初探

首先使用如下代码从 MovieLens dataset 官方网站下载电影评论数据集到本地，下载后会得到4个csv文件，分别为 links.csv、movies.csv、ratings.csv、tags.csv。

from torch_geometric.data import download_url

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/752621