Code for paper PairRE: Knowledge Graph Embeddings via Paired Relation Vectors.

Last update: Dec 19, 2022

Overview

PairRE

Code for paper PairRE: Knowledge Graph Embeddings via Paired Relation Vectors.

This implementation of PairRE for Open Graph Benchmak datasets (ogbl-wikikg and ogbl-biokg) is based on OGB. Thanks for their contributions.

Setup

To run the code, you need the following dependencies:

Pytorch 1.4.0
ogb 1.2.2 for wikikg
ogb 1.2.4 for wikikg2

Results

The results of PairRE on ogbl-wikikg, ogbl-biokg and ogbl-wikikg2 are as follows.

ogbl-wikikg

	#Dim	#Parameters	Hardware	Test MRR	Valid MRR
PairRE	100	250,167,400	16GB GPU	0.4912 ± 0.004	0.5013 ± 0.004
PairRE	200	500,334,800	16GB GPU	0.5289 ± 0.003	0.5529 ± 0.001

ogbl-biokg

	#Dim	#Parameters	Hardware	Test MRR	Valid MRR
PairRE	2000	187,750,000	16GB GPU	0.8164 ± 0.0005	0.8172 ± 0.0005

ogbl-wikikg2

	#Dim	#Parameters	Hardware	Test MRR	Valid MRR
PairRE	100	250,167,400	16GB GPU	0.4849 ± 0.003	0.4941 ± 0.004
PairRE	200	500,334,800	16GB GPU	0.5208 ± 0.003	0.5423 ± 0.002

Running the code

ogbl-wikikg

cd wikikg && sh examples.sh

ogbl-biokg

cd biokg && sh examples.sh

ogbl-wikikg2

Please update ogb package to version 1.2.4. The hyperparameters are same to the experiments in ogbl-wikikg.

cd wikikg && sh examples.sh

The details of the optional hyperparameters can be found in examples.sh.

Code for paper PairRE: Knowledge Graph Embeddings via Paired Relation Vectors.

Related tags

Overview

PairRE

Setup

Results

ogbl-wikikg

ogbl-biokg

ogbl-wikikg2

Running the code

ogbl-wikikg

ogbl-biokg

ogbl-wikikg2

Owner

Alipay

In this tutorial, you will perform inference across 10 well-known pre-trained object detectors and fine-tune on a custom dataset. Design and train your own object detector.

Lipstick ain't enough: Beyond Color-Matching for In-the-Wild Makeup Transfer (CVPR 2021)

Revisting Open World Object Detection

For auto aligning, cropping, and scaling HR and LR images for training image based neural networks

SpeechBrain is an open-source and all-in-one speech toolkit based on PyTorch.

darija <-> english dictionary

TSIT: A Simple and Versatile Framework for Image-to-Image Translation

ZeroVL - The official implementation of ZeroVL

Source code for Zalo AI 2021 submission

Using deep actor-critic model to learn best strategies in pair trading

Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecasting

Algebraic effect handlers in Python

Example of a Quantum LSTM

Official implementation of the ICCV 2021 paper "Joint Inductive and Transductive Learning for Video Object Segmentation"

Exploring Machine Learning Models for detecting anomalous behavior in credit-card transactions. It's crucial that credit-card companies are able to recognize fraudulent activity so that customers are not charged for items they didn't purchase.

MSG-Transformer: Exchanging Local Spatial Information by Manipulating Messenger Tokens

Data & Code for ACCENTOR Adding Chit-Chat to Enhance Task-Oriented Dialogues

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

基于AlphaPose的TensorRT加速

Code for the Convolutional Vision Transformer (ConViT)