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_1and
entity_2hidden state vectors, respectively. (including \$, # tokens)
--add_sep_tokenoption)
$ python3 main.py --do_train --do_eval
proposed_answers.txtin
evaldirectory.
$ python3 official_eval.py # macro-averaged F1 = 88.29%
Otherrelation)
result.txtin
evaldirectory.
$ python3 predict.py --input_file {INPUT_FILE_PATH} --output_file {OUTPUT_FILE_PATH} --model_dir {SAVED_CKPT_PATH}