KPRN - 知识图谱路径的推荐模型


MIT
跨平台
Python

软件简介

KPRN 由 eBay 开源,是一个知识图谱路径的推荐模型(Reasoning Over Knowledge Graph Paths for
Recommendation)。它广泛使用机器学习技术,有助于媒体等项目对推荐属性进行训练和预测,运行环境需配置 Python(2.7 / 3.5) 和
Lua(5.3)。

KPRN 是在 AAAI 2019 会议发表论文 “Explainable Reasoning over Knowledge Graphs for
Recommendation.”
的关联代码。该论文提出了一种基于 RNN
的推荐模型,对用户和物品之间的交互特征在知识图谱中存在的关联路径进行建模,为用户提供可解释性推荐。这个模型基于 LSTM
学习关联路径的表示,充分考虑了实体、关系间产生的序列依赖性,具备较强的推理能力。数据显示,该模型在电影数据集 MI 音乐数据集 KKBox
上取得了当前最优结果。