Bootstrap

动手学图神经网络(12):MovieLens上的链接回归

MovieLens上的链接回归

在MovieLens数据集上进行评分预测的实践过程,包括数据处理、模型构建、训练以及评估等步骤,预测用户对电影的评分(即边的属性值)。

环境设置

使用pip安装pyg - libpytorch_geometricsentence_transformersfuzzywuzzycaptum等。

import torch

print(torch.__version__)

import os

os.environ['TORCH'<
;