This is a Pytorch version of TreeGen (A Tree-Based Transformer Architecture for Code Generation).
Our paper is available at https://arxiv.org/abs/1911.09983. (Accepted by AAAI'20)
The model is in the hearthstone folder.
The preprocess is in the hearthstone_preprocess folder.
- NLTK 3.2.1
- Pytorch 1.10.0
- Python 3.7
- Ubuntu 16.04