/CDRGN-SDE

CDRGN-SDE: Cross-Dimensional Recurrent Graph Network with Neural Stochastic Differential Equation for Temporal Knowledge Graph Embedding

Primary LanguagePython

CDRGN-SDE: Cross-Dimensional Recurrent Graph Network with Neural Stochastic Differential Equation for Temporal Knowledge Graph Embedding

This repository provides the official PyTorch implementation of the research paper CDRGN-SDE: Cross-Dimensional Recurrent Graph Network with Neural Stochastic Differential Equation for Temporal Knowledge Graph Embedding. It will update soon. Thanks for your attention.

Requirements

  • python >= 3.6
  • torch >= 1.8.0
  • torchvision >= 0.9.2
  • torch-scatter >= 2.0.6
  • dgl-cu111 >= 0.6.1
  • tqdm
  • pandas
  • rdflib

How to Run

Step 1: pre-process the dataset

python data/dataset/ent2word.py
python get_neighbor.py

Step 2: training the model

python src/main.py