-
Notifications
You must be signed in to change notification settings - Fork 31
/
prepare_data.sh
71 lines (58 loc) · 3.06 KB
/
prepare_data.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
exp_dir=$1
src_lang=$2
tgt_lang=$3
train_data_dir=${4:-"$exp_dir/$src_lang-$tgt_lang"}
devtest_data_dir=${5:-"$exp_dir/devtest/all/$src_lang-$tgt_lang"}
echo "Running experiment ${exp_dir} on ${src_lang} to ${tgt_lang}"
train_processed_dir=$exp_dir/data
devtest_processed_dir=$exp_dir/data
out_data_dir=$exp_dir/final_bin
mkdir -p $train_processed_dir
mkdir -p $devtest_processed_dir
mkdir -p $out_data_dir
# train preprocessing
train_infname_src=$train_data_dir/train.$src_lang
train_infname_tgt=$train_data_dir/train.$tgt_lang
train_outfname_src=$train_processed_dir/train.SRC
train_outfname_tgt=$train_processed_dir/train.TGT
echo "Applying normalization and script conversion for train"
input_size=`python scripts/preprocess_translate.py $train_infname_src $train_outfname_src $src_lang`
input_size=`python scripts/preprocess_translate.py $train_infname_tgt $train_outfname_tgt $tgt_lang`
echo "Number of sentences in train: $input_size"
# dev preprocessing
dev_infname_src=$devtest_data_dir/dev.$src_lang
dev_infname_tgt=$devtest_data_dir/dev.$tgt_lang
dev_outfname_src=$devtest_processed_dir/dev.SRC
dev_outfname_tgt=$devtest_processed_dir/dev.TGT
echo "Applying normalization and script conversion for dev"
input_size=`python scripts/preprocess_translate.py $dev_infname_src $dev_outfname_src $src_lang`
input_size=`python scripts/preprocess_translate.py $dev_infname_tgt $dev_outfname_tgt $tgt_lang`
echo "Number of sentences in dev: $input_size"
# test preprocessing
test_infname_src=$devtest_data_dir/test.$src_lang
test_infname_tgt=$devtest_data_dir/test.$tgt_lang
test_outfname_src=$devtest_processed_dir/test.SRC
test_outfname_tgt=$devtest_processed_dir/test.TGT
echo "Applying normalization and script conversion for test"
input_size=`python scripts/preprocess_translate.py $test_infname_src $test_outfname_src $src_lang`
input_size=`python scripts/preprocess_translate.py $test_infname_tgt $test_outfname_tgt $tgt_lang`
echo "Number of sentences in test: $input_size"
echo "Learning bpe. This will take a very long time depending on the size of the dataset"
echo `date`
# learn bpe for preprocessed_train files
bash learn_bpe.sh $exp_dir
echo `date`
echo "Applying bpe"
bash apply_bpe_traindevtest_notag.sh $exp_dir
mkdir -p $exp_dir/final
# this is only required for joint training
# echo "Adding language tags"
# python scripts/add_tags_translate.py $outfname._bpe $outfname.bpe $src_lang $tgt_lang
# this is imporatnt step if you are training with tpu and using num_batch_buckets
# the currnet implementation does not remove outliers before bucketing and hence
# removing these large sentences ourselves helps with getting better buckets
python scripts/remove_large_sentences.py $exp_dir/bpe/train.SRC $exp_dir/bpe/train.TGT $exp_dir/final/train.SRC $exp_dir/final/train.TGT
python scripts/remove_large_sentences.py $exp_dir/bpe/dev.SRC $exp_dir/bpe/dev.TGT $exp_dir/final/dev.SRC $exp_dir/final/dev.TGT
python scripts/remove_large_sentences.py $exp_dir/bpe/test.SRC $exp_dir/bpe/test.TGT $exp_dir/final/test.SRC $exp_dir/final/test.TGT
echo "Binarizing data"
bash binarize_training_exp.sh $exp_dir SRC TGT