KazuhiraDZ / SE-CommentGeneration

source code for comment generation by a sequence to sequence attention model

TL-CodeSum

Dataset

The paper used two datasets:

-- Data for code summarization task

-- Data_pre for API sequence summarization task

Requirments

Python 3.6

NLTK 3.2.5

Config

The configuration of different models that we used is available in the folder "config"

Train

python3 main.py ../config/**.yaml --train -v

The project is based on the Projects: https://www.tensorflow.org/tutorials/seq2seq and https://github.com/eske/seq2seq