Advertisement

KGCN-pytorch:基于PyTorch的KGCN模型实现

  •  5星
  •     浏览量: 0
  •     大小:None
  •      文件类型:None


简介:
KGCN-pytorch是基于流行的深度学习框架PyTorch实现的知识图谱卷积网络(Knowledge Graph Convolutional Network, KGCN)模型。该库提供了灵活且高效的工具,用于处理和分析复杂的知识图谱数据,适用于推荐系统、问答系统等应用场景。 KGCN-火炬是推荐系统的知识图卷积网络的Pytorch实现:王宏伟、赵M、谢星、李文杰、郭敏仪在2019年网络会议论文集(WWW 2019)中的工作。 电影数据集的原始评级文件太大,无法包含在此仓库中。首先对评级数据进行分类: ``` wget http://files.grouplens.org/datasets/movielens/ml-20m.zip unzip ml-20m.zip mv ml-20m/ratings.csv data/movie/ ``` 音乐没事做其他数据集如果要使用自己的数据集,则需要准备两个文件。评分数据每行应包含(用户-ID、物品-ID和评级)。在此仓库中,它是pandas数据框结构。 知识图由每个三元组(头-关系尾)组成,在此仓库中,它们是字典类型。

全部评论 (0)

还没有任何评论哟~
客服
客服