There was an error while loading. Please reload this page.
1 parent 624eab6 commit 79f5891Copy full SHA for 79f5891
train.sh
@@ -1,3 +1,4 @@
1
+#!/usr/bin/env bash
2
###########################################################
3
# Change the following values to train a new model.
4
# type: the name of the new model, only affects the saved file name.
@@ -14,4 +15,6 @@ model_dir=models/${type}
14
15
16
mkdir -p models/${model_dir}
17
set -e
-python3 -u code2vec.py --data ${data} --test ${test_data} --save ${model_dir}/saved_model
18
+python3 -u code2vec.py --data ${data} --test ${test_data} --save ${model_dir}/saved_model
19
+
20
+python3 -u code2vec.py --data data/java14m/java14m --test data/java14m/java14m.val.c2v --save models/java14m/saved_model
0 commit comments