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

Code for the paper One Thing One Click: A Self-Training Approach for Weakly Supervised 3D Semantic Segmentation, CVPR 2021.

Official PyTorch implementation of "AASIST: Audio Anti-Spoofing using Integrated Spectro-Temporal Graph Attention Networks"

[CVPR 2021] Monocular depth estimation using wavelets for efficiency

PyTorch implementation of the WarpedGANSpace: Finding non-linear RBF paths in GAN latent space (ICCV 2021)

Fluency ENhanced Sentence-bert Evaluation (FENSE), metric for audio caption evaluation. And Benchmark dataset AudioCaps-Eval, Clotho-Eval.

DrWhy is the collection of tools for eXplainable AI (XAI). It's based on shared principles and simple grammar for exploration, explanation and visualisation of predictive models.

This repository contains the source code for the paper First Order Motion Model for Image Animation

😮The official implementation of "CoNeRF: Controllable Neural Radiance Fields" 😮

Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition

It is modified Tensorflow 2.x version of Mask R-CNN

Spam your friends and famly and when you do your famly will disown you and you will have no friends.

A light weight data augmentation tool for training CNNs and Viola Jones detectors

A set of simple scripts to process the Imagenet-1K dataset as TFRecords and make index files for NVIDIA DALI.

B2EA: An Evolutionary Algorithm Assisted by Two Bayesian Optimization Modules for Neural Architecture Search

Project repo for Learning Category-Specific Mesh Reconstruction from Image Collections

An end-to-end implementation of intent prediction with Metaflow and other cool tools

(ICCV 2021) ProHMR - Probabilistic Modeling for Human Mesh Recovery

Deep GPs built on top of TensorFlow/Keras and GPflow

TorchOk - The toolkit for fast Deep Learning experiments in Computer Vision

DANet for Tabular data classification/ regression.