Skip to content

msy0513/BiLSTM-and-CNN-for-Link-Prediction

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

36 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BiLSTM CNN for knowledge graph completion in Keras

Keras implementation for the paper titled "Path-based reasoning approach for knowledge graph completion using CNN-BiLSTM with attention mechanism". Given a candidate relation and two entities, it encodes paths that connect the entities into a low-dimensional space using a convolutional operation followed by BiLSTM. An attention layer is also applied to capture the semantic correlation between a candidate relation and each path between two entities and attentively extract reasoning evidence from the representation of multiple paths to predict whether the entities should be connected by the candidate relation.

Required Files

  • data/processed_data.tar.gz - dataset files containing grounded paths with relations and entities (e.g e1, r1, e2, r2, e3).
  • tasks - can be downloaded from [1].

For generating relation paths such as (r1, r2, ..., rk), we used [2].

The original knowledge graph data we used for our experiments can be found in [1] which includes tasks datasets with train/test triples.

If you use our code, please cite the paper

Batselem Jagvaral, Wan-Kon Lee, Jae-Seung Roh, Min-Sung Kim, Young-Tack Park, Path-based reasoning approach for knowledge graph completion using CNN-BiLSTM with attention mechanism, Expert Systems With Applications (2019), doi: https://doi.org/10.1016/j.eswa.2019.112960

Reference:

Acknowledgement

I'd like to thank my professor, Young-Tack Park for his support and advise.

About

Keras implementation of path-based link prediction model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 92.6%
  • Shell 7.4%