-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.sh
13 lines (9 loc) · 876 Bytes
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
#!/bin/sh
chmod +x auto2dep.sh
mkdir logs
mkdir models
mkdir tmp
# train on sample_data
python nest_ccg_main.py --do_train --tag_path=./sample_data/425tags --train_data_path=./sample_data/train.tsv --dev_data_path=./sample_data/dev.tsv --test_data_path=./sample_data/test.tsv --eval_data_dir=./sample_data/ --do_lower_case --use_bert --use_gcn --use_weight --use_in_chunk --use_cross_chunk --gcn_layer_number=2 --bert_model=/path/to/bert_base_uncased --max_seq_length=300 --train_batch_size=16 --eval_batch_size=16 --num_train_epochs=3 --warmup_proportion=0.1 --learning_rate=1e-5 --patient=15 --max_ngram_length=5 --ngram_freq_threshold=2 --model_name=test
# test on sample_data
python nest_ccg_main.py --do_test --tag_path=./sample_data/425tags --eval_data_path=./sample_data/test.tsv --eval_data_dir=./sample_data/ --eval_model=./models/model_name --eval_batch_size=16