BERT_models_v1

This repository contians the code and the data for the relation extraction work.

Running the code

The code can be run with pytorch and CUDA enabled.

.sent and .pointer files: The .sent files contain the sentences and the corresponding locations of entities and triples.

Converting into BERT embeddings: Run the code helper.py. The BERT embeddings will be generated.

Example: python3 helper.py train.sent train.pointer train_bert.sent train_bert.pointer train_bert.pos

Similarly for test and dev file.
Running the code: Run the code matbert_ptrnet_decoder.py for MatBERT. The parameters are as follows:

python3 [Program file] [CUDA device used] [seed] [job mode (train or test)] [batch size] [number of epoch] [type of nn (bi-directional or not etc.)] [order of triples (random or not)] [update BERT]

Example: python3 bert_ptrnet_decoder.py 0 1023 train 32 50 0 0 0 python3 bert_ptrnet_decoder.py 0 1023 test 32 50 0 0 0

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
README.md		README.md
bert_ptrnet_decoder.py		bert_ptrnet_decoder.py
dev.pointer		dev.pointer
dev.sent		dev.sent
dev_bert.pointer		dev_bert.pointer
dev_bert.pos		dev_bert.pos
dev_bert.sent		dev_bert.sent
helper.py		helper.py
matbert_ptrnet_decoder.py		matbert_ptrnet_decoder.py
relations.txt		relations.txt
roberta_ptrnet_decoder.py		roberta_ptrnet_decoder.py
test.pointer		test.pointer
test.sent		test.sent
test_bert.pointer		test_bert.pointer
test_bert.pos		test_bert.pos
test_bert.sent		test_bert.sent
test_new_bert.pos		test_new_bert.pos
train.pointer		train.pointer
train.sent		train.sent
train10.pointer		train10.pointer
train10.sent		train10.sent
train30.pointer		train30.pointer
train30.sent		train30.sent
train50.pointer		train50.pointer
train50.sent		train50.sent
train70.pointer		train70.pointer
train70.sent		train70.sent
train_10.pointer		train_10.pointer
train_10.sent		train_10.sent
train_5.pointer		train_5.pointer
train_5.sent		train_5.sent
train_bert.pointer		train_bert.pointer
train_bert.pos		train_bert.pos
train_bert.sent		train_bert.sent