/SG-Trans

Primary LanguagePython

Our code is adapted from https://github.com/wasiahmad/NeuralCodeSum

Training Models

You can train our model with following steps.

$ cd  DATASET_NAME/scripts/DATASET_NAME

choices for DATASET_NAME are ["java", "python"].

To train a model, run:

$ bash script_name.sh GPU_ID MODEL_NAME