Code for "JBNU-CCLab at SemEval-2022 Task 12: Machine Reading Comprehension and Span Pair Classification for Linking Mathematical Symbols to Their Descriptions", SemEval@NAACL2022 (1st at the all subtasks)
Paper - (https://aclanthology.org/2022.semeval-1.231/)
- PyTorch >= 1.7.1
- pytorch-lightning==0.9.0
- tokenizers==0.9.3
- pandas==1.3.3
- sklearn
- transformers==4.10.2
- Environment Setting
pip install -r ./NER/requirements.txt
pip install -r ./RE/requirements.txt
- Entity model(Train, infer)
bash ./NER/scripts/mrc_ner/reproduce/semeval.sh
bash ./NER/scripts/mrc_ner/nested_inference.sh
- Relation model(Train, infer, ensemble and post-process)
bash ./RE/run.sh base [bsz]
python ./RE/inference.py
python ./RE/get_ensemble_result.py
- Make submission
python ./make_result/make_result.py
https://drive.google.com/file/d/1SFm791Z7KW0GAZ45PyIak5xsL_LdVFOb/view?usp=sharing
If you encounter any problem, leave an issue in the github repo.