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 "Searching for Efficient Multi-Stage Vision Transformers"

PyTorch implementation of neural style randomization for data augmentation

Official PyTorch implementation of the ICRA 2021 paper: Adversarial Differentiable Data Augmentation for Autonomous Systems.

frida工具的缝合怪

[NeurIPS 2021] Large Scale Learning on Non-Homophilous Graphs: New Benchmarks and Strong Simple Methods

Python Environment for Bayesian Learning

Removing Inter-Experimental Variability from Functional Data in Systems Neuroscience

Simple keras FCN Encoder/Decoder model for MS-COCO (food subset) segmentation

Learning View Priors for Single-view 3D Reconstruction (CVPR 2019)

Predict Breast Cancer Wisconsin (Diagnostic) using Naive Bayes

Masked regression code - Masked Regression

[CIKM 2019] Code and dataset for "Fi-GNN: Modeling Feature Interactions via Graph Neural Networks for CTR Prediction"

Sample Code for "Pessimism Meets Invariance: Provably Efficient Offline Mean-Field Multi-Agent RL"

Implementing DropPath/StochasticDepth in PyTorch

Experiments with Fourier layers on simulation data.

Source code for The Power of Many: A Physarum Swarm Steiner Tree Algorithm

smc.covid is an R package related to the paper A sequential Monte Carlo approach to estimate a time varying reproduction number in infectious disease models: the COVID-19 case by Storvik et al

Code for "Diversity can be Transferred: Output Diversification for White- and Black-box Attacks"

FairMOT - A simple baseline for one-shot multi-object tracking

A Python library that enables ML teams to share, load, and transform data in a collaborative, flexible, and efficient way :chestnut: