上传者: 42166918
|
上传时间: 2021-12-13 11:43:44
|
文件大小: 4.35MB
|
文件类型: -
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/
音乐
没事做
其他数据集
如果要使用自己的数据集,则需要准备2个数据。
评分数据
每行应包含(user-item-rating)
在此回购中,它是pandas数据框结构。 (看看data_loader.py )
知识图
每个三元组(头-关系尾)由知识图组成
在此仓库中,它是字典类型。 (看看data_