IRIE_project_2018_relation_extraction

Relation Extraction with The Wall Street Journal

Prerequisite

Gensim
Keras
sklearn

workflow

Transfer documents into feature vectors by 3-billion words Google News pre-trained word2vec model
Train a Bidirectional LSTM following DNN model to predict Node Role/Label
Use the output of Bidirectional LSTM layer as input to train a DNN model for edge label prediction

model summary

node role prediction model

edge relation prediction model

result

source: https://arxiv.org/pdf/1809.02700.pdf