dataset_name=dataset
data_dir=../data/${dataset_name}
data=${data_dir}/${dataset_name}
test_data=${data_dir}/${dataset_name}.val.c2v
model_dir=models/${type}

mkdir -p ${model_dir}
set -e
python -u code2vec.py --data ${data} --save ${model_dir}/saved_model --test ${test_data}