tbcnn_testing_script.sh

PARSER=pycparser
TRAIN_PATH=/home/nghibui/codes/datasets/OJ_${PARSER}_train_test_val/${PARSER}-buckets-train.pkl
TEST_PATH=/home/nghibui/codes/datasets/OJ_${PARSER}_train_test_val/${PARSER}-buckets-test.pkl
NODE_TYPE_VOCABULARY_PATH=vocab/${PARSER}/node_type/type.txt
TOKEN_VOCABULARY_PATH=vocab/${PARSER}/node_token/token.txt
BATCH_SIZE=32
CHECKPOINT_EVERY=500
TREE_SIZE_THRESHOLD_UPPER=1500
TREE_SIZE_THRESHOLD_LOWER=0
CUDA=-1
VALIDATING=1
NODE_TYPE_DIM=100
NODE_TOKEN_DIM=100
CONV_OUTPUT_DIM=100
NUM_CONV=4
EPOCH=120
PYTHON=python3
NODE_INIT=2
BEST_F1=0.0
${PYTHON} test_tbcnn.py \
--train_path ${TRAIN_PATH} --test_path ${TEST_PATH} --batch_size ${BATCH_SIZE} \
--checkpoint_every ${CHECKPOINT_EVERY} --cuda ${CUDA} --validating ${VALIDATING} \
--tree_size_threshold_upper ${TREE_SIZE_THRESHOLD_UPPER} \
--tree_size_threshold_lower ${TREE_SIZE_THRESHOLD_LOWER} \
--node_type_dim ${NODE_TYPE_DIM} --node_token_dim ${NODE_TOKEN_DIM} \
--node_type_vocabulary_path ${NODE_TYPE_VOCABULARY_PATH} \
--token_vocabulary_path ${TOKEN_VOCABULARY_PATH} --epochs ${EPOCH} --parser ${PARSER} \
--node_init ${NODE_INIT} --num_conv ${NUM_CONV} --conv_output_dim ${CONV_OUTPUT_DIM}