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

A hybrid SOTA solution of LiDAR panoptic segmentation with C++ implementations of point cloud clustering algorithms. ICCV21, Workshop on Traditional Computer Vision in the Age of Deep Learning

Package for extracting emotions from social media text. Tailored for financial data.

[CVPR 2019 Oral] Multi-Channel Attention Selection GAN with Cascaded Semantic Guidance for Cross-View Image Translation

AI4Good project for detecting waste in the environment

ReAct: Out-of-distribution Detection With Rectified Activations

This library provides an abstraction to perform Model Versioning using Weight & Biases.

Using OpenAI's CLIP to upscale and enhance images

A simple version for graphfpn

Audio Source Separation is the process of separating a mixture into isolated sounds from individual sources

The Multi-Mission Maximum Likelihood framework (3ML)

Infrastructure as Code (IaC) for a self-hosted version of Gnosis Safe on AWS

Realtime_Multi-Person_Pose_Estimation

2021-AIAC-QQ-Browser-Hyperparameter-Optimization-Rank6

FPGA: Fast Patch-Free Global Learning Framework for Fully End-to-End Hyperspectral Image Classification

SysWhispers Shellcode Loader

Cmsc11 arcade - Final Project for CMSC11

This repository contains all code and data for the Inside Out Visual Place Recognition task

Flexible-Modal Face Anti-Spoofing: A Benchmark

Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.

PyTorch implementation of paper: HPNet: Deep Primitive Segmentation Using Hybrid Representations.