MovieLens上的链接回归
在MovieLens数据集上进行评分预测的实践过程,包括数据处理、模型构建、训练以及评估等步骤,预测用户对电影的评分(即边的属性值)。
环境设置
使用pip
安装pyg - lib
、pytorch_geometric
、sentence_transformers
、fuzzywuzzy
、captum
等。
import torch
print(torch.__version__)
import os
os.environ['TORCH'<