code2seq/train.sh at master · DeepLearnI/code2seq

executable file

19 lines (18 loc) · 805 Bytes

#!/usr/bin/env bash

###########################################################

# Change the following values to train a new model.

# type: the name of the new model, only affects the saved file name.

# dataset: the name of the dataset, as was preprocessed using preprocess.sh

# test_data: by default, points to the validation set, since this is the set that

# will be evaluated after each training iteration. If you wish to test

# on the final (held-out) test set, change 'val' to 'test'.

type=python_20k

dataset_name=python_20k

data_dir=data/${dataset_name}

data=${data_dir}/${dataset_name}

test_data=${data_dir}/${dataset_name}.val.c2s

model_dir=models/${type}

PYTHON=python

mkdir -p ${model_dir}

set -e

${PYTHON} -u code2seq.py --data ${data} --test ${test_data} --save_prefix ${model_dir}/model

Provide feedback