PGPR:推荐知识的增强知识图推理

【文件属性】：

文件名称：PGPR:推荐知识的增强知识图推理

文件大小：63.58MB

文件格式：ZIP

更新时间：2024-05-24 12:27:28

Python

推荐知识的增强知识图推理 该存储库包含SIGIR 2019论文“” [2]的源代码。 数据集 “ data /”目录中有两个Amazon数据集（Amazon_Beauty，Amazon_Cellphones），并且拆分与[1]一致。 本文中使用的所有四个数据集都可以下载。 要求 Python> = 3.6 PyTorch = 1.0 如何运行代码 首先处理数据： python preprocess.py --dataset < dataset> “ ”应该是“ cd”，“ beauty”，“ cloth”，“ cell”之一（请参阅utils.py）。 训练知识图嵌入（在这种情况下为TransE）： python train_transe_model.py --dataset < dataset> 训练RL代理商： pyth

【文件预览】：
PGPR-master
----knowledge_graph.py(9KB)
----train_transe_model.py(6KB)
----transe_model.py(10KB)
----data_utils.py(9KB)
----utils.py(6KB)
----train_agent.py(9KB)
----requirements.txt(354B)
----test_agent.py(10KB)
----README.md(2KB)
----data()
--------Amazon_Beauty()
--------Amazon_Cellphones()
----.gitignore(1KB)
----preprocess.py(2KB)
----kg_env.py(11KB)