Context-aware Entity Typing in Knowledge Graphs

This is the source code for: Context-aware Entity Typing in Knowledge Graphs.

Requirements

Python 3
PyTorch >= 1.6.0
dgl >= 0.5.3

Usage

Data preprocessing:

cd data
python preprocess.py --dataset FB15kET
python preprocess.py --dataset YAGO43kET

Train:

########### FB15kET ###########
# CET
python run.py --model CET --dataset FB15kET --load_ET --load_KG --neighbor_sampling \
--hidden_dim 100 --temperature 0.5 --lr 0.001 --loss FNA --beta 4.0 --cuda

# R-GCN
python run.py --model RGCN --dataset FB15kET --load_ET --load_KG --neighbor_sampling \
--hidden_dim 100 --lr 0.001 --loss FNA --beta 3.0 --cuda

# CompGCN
python run.py --model CompGCN --dataset FB15kET --load_ET --load_KG --neighbor_sampling \
--hidden_dim 100 --lr 0.001 --loss FNA --activation relu --cuda

########### YAGO43kET ###########
# CET
python run.py --model CET --dataset YAGO43kET --load_ET --load_KG --neighbor_sampling \
--hidden_dim 100 --temperature 0.5 --lr 0.001 --loss FNA --beta 2.0 --cuda

# R-GCN
python run.py --model RGCN --dataset YAGO43kET --load_ET --load_KG --neighbor_sampling \
--hidden_dim 100 --lr 0.001 --loss FNA --beta 2.0 --cuda

# CompGCN
python run.py --model CompGCN --dataset YAGO43kET --load_ET --load_KG --neighbor_sampling \
--hidden_dim 100 --lr 0.001 --loss FNA --activation relu --cuda

The KGE baselines can be found in KGE_baselines.

Acknowledgement

We refer to the code of DGL. Thanks for their contributions.

This is the source code for: Context-aware Entity Typing in Knowledge Graphs.

Related tags

Overview

Context-aware Entity Typing in Knowledge Graphs

Requirements

Usage

Acknowledgement

Owner

Official PyTorch implementation of "Improving Face Recognition with Large AgeGaps by Learning to Distinguish Children" (BMVC 2021)

DALL-Eval: Probing the Reasoning Skills and Social Biases of Text-to-Image Generative Transformers

A tutorial showing how to train, convert, and run TensorFlow Lite object detection models on Android devices, the Raspberry Pi, and more!

Code repository for the paper "Tracking People with 3D Representations"

A pytorch implementation of MBNET: MOS PREDICTION FOR SYNTHESIZED SPEECH WITH MEAN-BIAS NETWORK

🛰️ List of earth observation companies and job sites

Official TensorFlow code for the forthcoming paper

An improvement of FasterGICP: Acceptance-rejection Sampling based 3D Lidar Odometry

VisualGPT: Data-efficient Adaptation of Pretrained Language Models for Image Captioning

Pre-trained model, code, and materials from the paper "Impact of Adversarial Examples on Deep Learning Models for Biomedical Image Segmentation" (MICCAI 2019).

Clairvoyance: a Unified, End-to-End AutoML Pipeline for Medical Time Series

Object DGCNN and DETR3D, Our implementations are built on top of MMdetection3D.

EdiBERT, a generative model for image editing

Towards Open-World Feature Extrapolation: An Inductive Graph Learning Approach

Voice Conversion by CycleGAN (语音克隆/语音转换)：CycleGAN-VC3

An open-source online reverse dictionary.

Speeding-Up Back-Propagation in DNN: Approximate Outer Product with Memory

Deduplicating Training Data Makes Language Models Better

Code for "Solving Graph-based Public Good Games with Tree Search and Imitation Learning"

[CVPR 2016] Unsupervised Feature Learning by Image Inpainting using GANs