bigdata-ustc / hms

HMS: A Hierarchical Solver with Dependency-Enhanced Understanding for Math Word Problem

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.

Dependencies

• python >= 3.6

• stanfordcorenlp

• torch

Usage

Preprocess dataset

python3 src/dataprocess/math23k.py

Train and test model

python3 src/main.py

For running arguments, please refer to src/config.py.

Citation

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}
}