/TreeGen-Pytorch

Primary LanguagePythonMIT LicenseMIT

TreeGen

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.

Dependenices

  • NLTK 3.2.1
  • Pytorch 1.10.0
  • Python 3.7
  • Ubuntu 16.04