Skip to content

Commit

Permalink
init
Browse files Browse the repository at this point in the history
  • Loading branch information
jasonwu0731 committed Jun 19, 2020
1 parent 7974c60 commit e1c6c76
Show file tree
Hide file tree
Showing 69 changed files with 6,267 additions and 0 deletions.
137 changes: 137 additions & 0 deletions evaluation_pipeline.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,137 @@
gpu=$1
model=$2
bert_dir=$3
output_dir=$4
add1=$5
add2=$6
add3=$7

# ./evaluation_pipeline.sh 0 bert bert-base-uncased save/BERT

# Intent
# for bsz in 8 16 32
# do
# CUDA_VISIBLE_DEVICES=$gpu python main.py \
# --my_model=multi_class_classifier \
# --dataset='["oos_intent"]' \
# --task_name="intent" \
# --earlystop="acc" \
# --output_dir=${output_dir}/Intent/OOS/BSZ${bsz} \
# --do_train \
# --task=nlu \
# --example_type=turn \
# --model_type=${model} \
# --model_name_or_path=${bert_dir} \
# --batch_size=${bsz} \
# --usr_token=[USR] --sys_token=[SYS] \
# --epoch=50 --eval_by_step=500 --warmup_steps=250 \
# $add1 $add2 $add3
# done

# DST
# CUDA_VISIBLE_DEVICES=$gpu python main.py \
# --my_model=BeliefTracker \
# --model_type=${model} \
# --dataset='["multiwoz"]' \
# --task_name="dst" \
# --earlystop="joint_acc" \
# --output_dir=${output_dir}/DST/MWOZ \
# --do_train \
# --task=dst \
# --example_type=turn \
# --model_name_or_path=${bert_dir} \
# --batch_size=6 --eval_batch_size=6 \
# --usr_token=[USR] --sys_token=[SYS] \
# --eval_by_step=4000 \
# $add1 $add2 $add3

# DA
CUDA_VISIBLE_DEVICES=$gpu python main.py \
--my_model=multi_label_classifier \
--do_train --dataset='["multiwoz"]' \
--task=dm --task_name=sysact --example_type=turn \
--model_type=${model} \
--model_name_or_path=${bert_dir} \
--output_dir=${output_dir}/DA/MWOZ/BSZ8 \
--batch_size=8 \
--eval_batch_size=4 \
--learning_rate=5e-5 \
--eval_by_step=1000 \
--usr_token=[USR] --sys_token=[SYS] \
--earlystop=f1_weighted \
$add1 $add2 $add3

CUDA_VISIBLE_DEVICES=$gpu python main.py \
--my_model=multi_label_classifier \
--do_train \
--dataset='["universal_act_dstc2"]' \
--task=dm --task_name=sysact --example_type=turn \
--model_type=${model} --model_name_or_path=${bert_dir} \
--output_dir=${output_dir}/DA/DSTC2/BSZ8 \
--batch_size=8 \
--eval_batch_size=4 \
--learning_rate=5e-5 \
--eval_by_step=500 \
--usr_token=[USR] --sys_token=[SYS] \
--earlystop=f1_weighted \
$add1 $add2 $add3

CUDA_VISIBLE_DEVICES=$gpu python main.py \
--my_model=multi_label_classifier \
--do_train \
--dataset='["universal_act_sim_joint"]' \
--task=dm --task_name=sysact --example_type=turn \
--model_type=${model} --model_name_or_path=${bert_dir} \
--output_dir=${output_dir}/DA/SIM_JOINT/BSZ8 \
--batch_size=8 \
--eval_batch_size=4 \
--learning_rate=5e-5 \
--eval_by_step=500 \
--usr_token=[USR] --sys_token=[SYS] \
--earlystop=f1_weighted \
$add1 $add2 $add3

# Response Selection
CUDA_VISIBLE_DEVICES=$gpu python main.py \
--my_model=dual_encoder_ranking \
--do_train \
--task=nlg \
--task_name=rs \
--example_type=turn \
--model_type=${model} \
--model_name_or_path=${bert_dir} \
--output_dir=${output_dir}/RS/MWOZ/ \
--batch_size=25 --eval_batch_size=100 \
--usr_token=[USR] --sys_token=[SYS] \
--fix_rand_seed \
--eval_by_step=1000 \
--max_seq_length=256 \
$add1 $add2 $add3

CUDA_VISIBLE_DEVICES=$gpu python main.py \
--my_model=dual_encoder_ranking \
--do_train \
--dataset='["universal_act_dstc2"]' \
--task=nlg --task_name=rs \
--example_type=turn \
--model_type=${model} \
--model_name_or_path=${bert_dir} \
--output_dir=${output_dir}/RS/DSTC2/ \
--batch_size=25 --eval_batch_size=100 \
--max_seq_length=256\
--fix_rand_seed \
$add1 $add2 $add3

CUDA_VISIBLE_DEVICES=$gpu python main.py \
--my_model=dual_encoder_ranking \
--do_train \
--dataset='["universal_act_sim_joint"]' \
--task=nlg --task_name=rs \
--example_type=turn \
--model_type=${model} \
--model_name_or_path=${bert_dir} \
--output_dir=${output_dir}/RS/SIM_JOINT/ \
--batch_size=25 --eval_batch_size=100 \
--max_seq_length=256 \
--fix_rand_seed \
$add1 $add2 $add3
Loading

0 comments on commit e1c6c76

Please sign in to comment.