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

Referring Video Object Segmentation

Repo for my Tensorflow/Keras CV experiments. Mostly revolving around the Danbooru20xx dataset

Metric learning algorithms in Python

Introducing neural networks to predict stock prices

Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"

Code for the paper "There is no Double-Descent in Random Forests"

Latex code for making neural networks diagrams

A lightweight face-recognition toolbox and pipeline based on tensorflow-lite

Pytorch implementation of AREL

Single-stage Keypoint-based Category-level Object Pose Estimation from an RGB Image

This app is a simple example of using Strealit to create a financial data web app.

Tensorflow 2 implementation of our high quality frame interpolation neural network

Self-supervised Label Augmentation via Input Transformations (ICML 2020)

Official PyTorch implementation of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image", ICCV 2019

Enabling dynamic analysis of Legacy Embedded Systems in full emulated environment

OptNet: Differentiable Optimization as a Layer in Neural Networks

ICRA 2021 - Robust Place Recognition using an Imaging Lidar

Deep Networks with Recurrent Layer Aggregation

This repository provides some of the code implemented and the data used for the work proposed in "A Cluster-Based Trip Prediction Graph Neural Network Model for Bike Sharing Systems".

Background Matting: The World is Your Green Screen