output_dir="./saved_models/bears_bugs"
train_data_file="./data/processed/bears_and_bugs/train.jsonl"
eval_data_file="./data/processed/bears_and_bugs/valid.jsonl"
test_data_file="./data/processed/bears_and_bugs/test.jsonl"
word2vec_path="./saved_models/word2vec/java_word2vec/word2vec_model.bin"

mkdir -p "$output_dir"

python run.py --lang java \
    --train_data_file $train_data_file \
    --eval_data_file $eval_data_file \
    --test_data_file $test_data_file \
    --seed 38 \
    --word2vec_path $word2vec_path \
    --block_size 400 --num_nodes 200 \
    --do_train --do_eval --do_test --evaluate_during_training \
    --train_batch_size 16 --eval_batch_size 16 --test_batch_size 16 \
    --learning_rate 1e-4 --learning_decay 0.9999999 --weight_decay 0.01 \
    --do_use_best_f1 --dynamic_thresholds \
    --epochs 200 --pre_epoch_eval 1 \
    --model ERUMGraphRegressionModel \
    --output_dir $output_dir \
    --feature_dim_size 129 --hidden_size 256 \
    --num_samples 5 --walk_length 5 --depth 3 --num_layers 2 --num_classes 2 \
    --directed 2>&1 | tee "${output_dir}/training.log"
