Pytorch implementation of R-BERT: "Enriching Pre-trained Language Model with Entity Information for Relation Classification"
(Unofficial) Pytorch implementation of R-BERT
: Enriching Pre-trained Language Model with Entity Information for Relation Classification
entity_1
and entity_2
hidden state vectors, respectively. (including $, # tokens)--add_sep_token
option)$ python3 main.py --do_train --do_eval
proposed_answers.txt
in eval
directory.$ python3 official_eval.py
# macro-averaged F1 = 88.29%
Other
relation)result.txt
in eval
directory.$ python3 predict.py --input_file {INPUT_FILE_PATH} --output_file {OUTPUT_FILE_PATH} --model_dir {SAVED_CKPT_PATH}