Leveraging Grammar and Reinforcement Learning for Neural Program Synthesis

TensorFlow implementation of Leveraging Grammar and Reinforcement Learning for Neural Program Synthesis.

Requirements

• Python 2.7+
• tqdm
• karel
• TensorFlow 1.4.1

Usage

Prepare with:

$ pip install -r requirements.txt

To generate datasets:

$ python dataset.py --data_dir=data --max_depth=5

To train a model:

$ python main.py
$ tensorboard --logdir=logs --host=0.0.0.0

Results

Currently, only maximum likelihood optimization is implemented. Expected correctness and RL is in progress.

(in progress)

References

• Karel dataset
• Neural Program Meta-Induction

Author

Taehoon Kim / @carpedm20