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

Nvdiffrast - Modular Primitives for High-Performance Differentiable Rendering

GPT-Code-Clippy (GPT-CC) is an open source version of GitHub Copilot

Convolutional Neural Network for Text Classification in Tensorflow

Online-compatible Unsupervised Non-resonant Anomaly Detection Repository

📚 A collection of all the Deep Learning Metrics that I came across which are not accuracy/loss.

source code of Adversarial Feedback Loop Paper

This is the official released code for our paper, The Emergence of Objectness: Learning Zero-Shot Segmentation from Videos

Pytorch implementation for "Distribution-Balanced Loss for Multi-Label Classification in Long-Tailed Datasets" (ECCV 2020 Spotlight)

the code for our CVPR 2021 paper Bilateral Grid Learning for Stereo Matching Network [BGNet]

Strongly local p-norm-cut algorithms for semi-supervised learning and local graph clustering

Code for Temporally Abstract Partial Models

object detection; robust detection; ACM MM21 grand challenge; Security AI Challenger Phase VII

Implementation for Shape from Polarization for Complex Scenes in the Wild

Continual reinforcement learning baselines: experiment specifications, implementation of existing methods, and common metrics. Easily extensible to new methods.

JAX + dataclasses

Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.

This project aims to segment 4 common retinal lesions from Fundus Images.

Aquarius - Enabling Fast, Scalable, Data-Driven Virtual Network Functions

Trained on Simulated Data, Tested in the Real World

Find the Heart simple Python Game