Adding code

nusnlp · Aug 7, 2021 · c65c5c9 · c65c5c9
commit c65c5c9
Show file tree

Hide file tree

Showing 2 changed files with 2,114 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -0,0 +1,24 @@
+This repository contains the source code of the paper "A Hierarchical Entity Graph Convolutional Network for Multi-hop Relation Extraction across Documents" published in RANLP 2021.
+
+### Datasets ###
+
+The dataset used for experiments in the paper can be downloaded from the following link:
+
+https://drive.google.com/drive/folders/1zSlXoeppoNpihbN75JbxuqZWiiYbpCLM
+
+### Requirements ###
+
+1) python3.6
+2) pytorch 1.7
+3) CUDA 8.0
+
+### How to run ###
+
+python3.6 models.py source_dir embedding_file target_dir model_id train
+
+python3.6 models.py source_dir embedding_file target_dir model_id test threshold
+
+Use model_id as 1 for CNN, 2 for BiLSTM, 3 for BiLSTM_CNN, 4 for LinkPath, and 5 for our HEGCN model.
+
+
+