Source code for paper HMS: A Hierarchical Solver with Dependency-Enhanced Understanding for Math Word Problem. An extended version of the work is available here.
-
python >= 3.6
-
stanfordcorenlp
-
torch
Preprocess dataset
python3 src/dataprocess/math23k.py
Train and test model
python3 src/main.py
For running arguments, please refer to src/config.py.
If you find our work helpful, please consider citing our paper.
@inproceedings{lin2021hms,
title={HMS: A Hierarchical Solver with Dependency-Enhanced Understanding for Math Word Problem},
author={Lin, Xin and Huang, Zhenya and Zhao, Hongke and Chen, Enhong and Liu, Qi and Wang, Hao and Wang, Shijin},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={35},
number={5},
pages={4232--4240},
year={2021}
}