The code is the implementation of PseDual for reexamination.
- Pytorch = 1.6.0;
- Transformers
Math23k and MathQA datasets can be downloaded from public sources and set to ./data directory.
The annotations include problem tokens, expression, num_list and nums position in problem.
Pre-trained bert-base model for problems encoding can be downloaded to a ./pretrained_models directory.
The main.py can be used to train and test model. The example trained model can be downloaded.
python main.py