/HFMRE

HFMRE: Constructing Huffman Tree in Bags to Find Excellent Instances for Distantly Supervised Relation Extraction

Primary LanguagePython

HFMRE: Constructing Huffman Tree in Bags to Find Excellent Instances for Distantly Supervised Relation Extraction

Reqirements

torch==1.6.0

nltk==3.6.2

numpy==1.19.5

scikit-learn==0.24.2

scipy==1.5.4

sentencepiece==0.1.95

transformers==3.0.2

tokenizers==0.8.1rc1

gdown==4.2.0

Datasets:

download_nyt10.sh

download_nyt10m.sh

download_wiki20m.sh

Train:

train_nyt10d.sh

train_nyt10m.sh

train_wiki20m.sh