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

Simple renderer for use with MuJoCo (>=2.1.2) Python Bindings.

This is the formal code implementation of the CVPR 2022 paper 'Federated Class Incremental Learning'.

Accurate 3D Face Reconstruction with Weakly-Supervised Learning: From Single Image to Image Set (CVPRW 2019). A PyTorch implementation.

Personal thermal comfort models using digital twins: Preference prediction with BIM-extracted spatial-temporal proximity data from Build2Vec

This repo will contain code to reproduce and build upon understanding transfer learning

we propose EfficientDerain for high-efficiency single-image deraining

[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations

Revitalizing CNN Attention via Transformers in Self-Supervised Visual Representation Learning

TAP: Text-Aware Pre-training for Text-VQA and Text-Caption, CVPR 2021 (Oral)

IMBENS: class-imbalanced ensemble learning in Python.

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

MazeRL is an application oriented Deep Reinforcement Learning (RL) framework

Few-NERD: Not Only a Few-shot NER Dataset

Evaluating different engineering tricks that make RL work

Code for WECHSEL: Effective initialization of subword embeddings for cross-lingual transfer of monolingual language models.

Pip-package for trajectory benchmarking from "Be your own Benchmark: No-Reference Trajectory Metric on Registered Point Clouds", ECMR'21

Image inpainting using Gaussian Mixture Models

Compare outputs between layers written in Tensorflow and layers written in Pytorch

Stochastic Extragradient: General Analysis and Improved Rates