Pgpr

Reinforcement Knowledge Graph Reasoning for Explainable Recommendation
Alternatives To Pgpr
Project NameStarsDownloadsRepos Using ThisPackages Using ThisMost Recent CommitTotal ReleasesLatest ReleaseOpen IssuesLicenseLanguage
Fl_chart5,9067166 days ago87June 10, 2023245mitDart
FL Chart is a highly customizable Flutter chart library that supports Line Chart, Bar Chart, Pie Chart, Scatter Chart, and Radar Chart.
Ogb1,721222 months ago19November 02, 202219mitPython
Benchmark datasets, data loaders, and evaluators for graph machine learning
Cogdl1,43715 months ago17June 01, 202234mitPython
CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)
Graph2vec791
a year ago1gpl-3.0Python
A parallel implementation of "graph2vec: Learning Distributed Representations of Graphs" (MLGWorkshop 2017).
Stgcn_ijcai 18723
7 months ago9bsd-2-clausePython
[IJCAI'18] Spatio-Temporal Graph Convolutional Networks
Knowledge Graph Learning662
8 months ago336mit
A curated list of awesome knowledge graph tutorials, projects and communities.
Sr Gnn607
2 years ago6Python
[AAAI 2019] Source code and datasets for "Session-based Recommendation with Graph Neural Networks"
Conve574
a year ago22mitPython
Convolutional 2D Knowledge Graph Embeddings resources
Knowledge_graph_attention_network434
3 years ago24mitPython
KGAT: Knowledge Graph Attention Network for Recommendation, KDD2019
Cleora434
5 months ago12otherJupyter Notebook
Cleora AI is a general-purpose model for efficient, scalable learning of stable and inductive entity embeddings for heterogeneous relational data.
Alternatives To Pgpr
Select To Compare


Alternative Project Comparisons
Readme

Reinforcement Knowledge Graph Reasoning for Explainable Recommendation

This repository contains the source code of the SIGIR 2019 paper "Reinforcement Knowledge Graph Reasoning for Explainable Recommendation" [2].

Datasets

Two Amazon datasets (Amazon_Beauty, Amazon_Cellphones) are available in the "data/" directory and the split is consistent with [1]. All four datasets used in this paper can be downloaded here.

Requirements

  • Python >= 3.6
  • PyTorch = 1.0

How to run the code

  1. Proprocess the data first:
python preprocess.py --dataset <dataset_name>

"<dataset_name>" should be one of "cd", "beauty", "cloth", "cell" (refer to utils.py).

  1. Train knowledge graph embeddings (TransE in this case):
python train_transe_model.py --dataset <dataset_name>
  1. Train RL agent:
python train_agent.py --dataset <dataset_name>
  1. Evaluation
python test_agent.py --dataset <dataset_name> --run_path True --run_eval True

If "run_path" is True, the program will generate paths for recommendation according to the trained policy. If "run_eval" is True, the program will evaluate the recommendation performance based on the resulting paths.

References

[1] Yongfeng Zhang, Qingyao Ai, Xu Chen, W. Bruce Croft. "Joint Representation Learning for Top-N Recommendation with Heterogeneous Information Sources". In Proceedings of CIKM. 2017.

[2] Yikun Xian, Zuohui Fu, S. Muthukrishnan, Gerard de Melo, Yongfeng Zhang. "Reinforcement Knowledge Graph Reasoning for Explainable Recommendation." In Proceedings of SIGIR. 2019.

Popular Graph Projects
Popular Dataset Projects
Popular Computer Science Categories

Get A Weekly Email With Trending Projects For These Categories
No Spam. Unsubscribe easily at any time.
Python
Dataset
Graph
Amazon