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

The materials used in the SaxonJS tutorial presented at Declarative Amsterdam, 2021

DatasetGAN: Efficient Labeled Data Factory with Minimal Human Effort

BalaGAN: Image Translation Between Imbalanced Domains via Cross-Modal Transfer

Company clustering with K-means/GMM and visualization with PCA, t-SNE, using SSAN relation extraction

Summary Explorer is a tool to visually explore the state-of-the-art in text summarization.

Adaptive Graph Convolution for Point Cloud Analysis

Diffusion Probabilistic Models for 3D Point Cloud Generation (CVPR 2021)

Patch SVDD for Image anomaly detection

[NeurIPS 2021] A weak-shot object detection approach by transferring semantic similarity and mask prior.

R-package accompanying the paper "Dynamic Factor Model for Functional Time Series: Identification, Estimation, and Prediction"

Using OpenAI's CLIP to upscale and enhance images

Parasite: a tool allowing you to compress and decompress files, to reduce their size

Code for "Neural 3D Scene Reconstruction with the Manhattan-world Assumption" CVPR 2022 Oral

An efficient and easy-to-use deep learning model compression framework

Adversarial Adaptation with Distillation for BERT Unsupervised Domain Adaptation

Deep Reinforcement Learning based autonomous navigation for quadcopters using PPO algorithm.

Get a Grip! - A robotic system for remote clinical environments.

Code for ICML 2021 paper: How could Neural Networks understand Programs?

This repo is official PyTorch implementation of MobileHumanPose: Toward real-time 3D human pose estimation in mobile devices(CVPRW 2021).

StyleGAN2-ADA - Official PyTorch implementation