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

Functional deep learning

Aquarius - Enabling Fast, Scalable, Data-Driven Virtual Network Functions

Reference implementation for Structured Prediction with Deep Value Networks

Plaything for Autistic Children (demo for PaddlePaddle/Wechaty/Mixlab project)

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

Human annotated noisy labels for CIFAR-10 and CIFAR-100.

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Official repository of "Investigating Tradeoffs in Real-World Video Super-Resolution"

StyleGAN-Human: A Data-Centric Odyssey of Human Generation

StocksMA is a package to facilitate access to financial and economic data of Moroccan stocks.

Tensorflow-Project-Template - A best practice for tensorflow project template architecture.

PyTorch implementation of "Simple and Deep Graph Convolutional Networks"

The code release of paper Low-Light Image Enhancement with Normalizing Flow

Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight

Code repository for the work "Multi-Domain Incremental Learning for Semantic Segmentation", accepted at WACV 2022

[AAAI2021] The source code for our paper 《Enhancing Unsupervised Video Representation Learning by Decoupling the Scene and the Motion》.

imbalanced-DL: Deep Imbalanced Learning in Python

An implementation of MobileFormer

A PyTorch Implementation of Single Shot MultiBox Detector

A robust camera and Lidar fusion based velocity estimator to undistort the pointcloud.