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

PyTea: PyTorch Tensor shape error analyzer

A clean and extensible PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

Perturb-and-max-product: Sampling and learning in discrete energy-based models

Code for CVPR2019 Towards Natural and Accurate Future Motion Prediction of Humans and Animals

Code for paper "A Critical Assessment of State-of-the-Art in Entity Alignment" (https://arxiv.org/abs/2010.16314)

An atmospheric growth and evolution model based on the EVo degassing model and FastChem 2.0

Python scripts for performing stereo depth estimation using the MobileStereoNet model in Tensorflow Lite.

Planner_backend - Academic planner application designed for students and counselors.

Self-Supervised CNN-GCN Autoencoder

Spiking Neural Network for Computer Vision using SpikingJelly framework and Pytorch-Lightning

Robust Consistent Video Depth Estimation

Learning to Reach Goals via Iterated Supervised Learning

Implementation of the state of the art beat-detection, downbeat-detection and tempo-estimation model

This repository is dedicated to developing and maintaining code for experiments with wide neural networks.

Pytorch implementation of SimSiam Architecture

Grammar Induction using a Template Tree Approach

Explainer for black box models that predict molecule properties

YOLOX-CondInst - Implement CondInst which is a instances segmentation method on YOLOX

Pytorch implementation for the EMNLP 2020 (Findings) paper: Connecting the Dots: A Knowledgeable Path Generator for Commonsense Question Answering

Torchlight2 lan game server tool - A message forwarding tool for Torchlight 2 lan game