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

Blender Add-On for slicing meshes with planes

This repository contains the code for the ICCV 2019 paper "Occupancy Flow - 4D Reconstruction by Learning Particle Dynamics"

A flexible framework of neural networks for deep learning

Implementation of Feedback Transformer in Pytorch

Code for generating a single image pretraining dataset

Computationally Efficient Optimization of Plackett-Luce Ranking Models for Relevance and Fairness

Solution to the first stage Quiz of Hamoye internship: Introduction to Python for Machine Learning

Official implementation of MSR-GCN (ICCV 2021 paper)

YuNetのPythonでのONNX、TensorFlow-Lite推論サンプル

[ACMMM 2021, Oral] Code release for "Elastic Tactile Simulation Towards Tactile-Visual Perception"

DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsification

Elevation Mapping on GPU.

🤖 A Python library for learning and evaluating knowledge graph embeddings

CondenseNet V2: Sparse Feature Reactivation for Deep Networks

UA-GEC: Grammatical Error Correction and Fluency Corpus for the Ukrainian Language

Implementation of MA-Trace - a general-purpose multi-agent RL algorithm for cooperative environments.

Generate images from texts. In Russian. In PaddlePaddle

Springer Link Download Module for Python

This repository contains the implementations related to the experiments of a set of publicly available datasets that are used in the time series forecasting research space.

This repository holds code and data for our PETS'22 article 'From "Onion Not Found" to Guard Discovery'.