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

Data augmentation for NLP, accepted at EMNLP 2021 Findings

Blender scripts for computing geodesic distance

Unofficial implement with paper SpeakerGAN: Speaker identification with conditional generative adversarial network

AWS provides a Python SDK, "Boto3" ,which can be used to access the AWS-account from the local.

Evaluation toolkit of the informative tracking benchmark comprising 9 scenarios, 180 diverse videos, and new challenges.

Tensorflow Tutorials using Jupyter Notebook

An easy-to-use app to visualise attentions of various VQA models.

CTF challenges from redpwnCTF 2021

Learning hidden low dimensional dyanmics using a Generalized Onsager Principle and neural networks

The repo of Feedback Networks, CVPR17

RoboDesk A Multi-Task Reinforcement Learning Benchmark

Code to train models from "Paraphrastic Representations at Scale".

LibFewShot: A Comprehensive Library for Few-shot Learning.

A Pytorch implementation of CVPR 2021 paper "RSG: A Simple but Effective Module for Learning Imbalanced Datasets"

MonoScene: Monocular 3D Semantic Scene Completion

Official code repository for the EMNLP 2021 paper

Unofficial PyTorch implementation of Guided Dropout

This repository is the official implementation of Open Rule Induction. This paper has been accepted to NeurIPS 2021.

Unofficial implementation of Google "CutPaste: Self-Supervised Learning for Anomaly Detection and Localization" in PyTorch

Built a deep neural network (DNN) that functions as an end-to-end machine translation pipeline