prepare datasets

$ mkdir data
$ wget https://s3.amazonaws.com/code2vec/data/java-large_data.tar.gz
$ tar -xvzf java-large_data.tar.gz
$ mv java-large data

train

for local testing

$ python code2vec.py --train --batch-size 2

for training

$ nohup python code2vec.py --train &

predict

$ wget https://github.com/tech-srl/code2vec/raw/master/JavaExtractor/JPredict/target/JavaExtractor-0.0.1-SNAPSHOT.jar

$ python code2vec --predict
$ python code2vec --predict --file Hoge.java --model hoge.ckpt