File size: 1,049 Bytes
2188a91 | 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 | #!/bin/bash
# Claims 3 & 4: ADFTD ablation (Scratch / Pre-trained L_rec / Pre-trained L_rec+L_div)
# and the Ti-MAE / SimMTM fine-tuning-promotion comparison of Table 4.
set -u
cd /home/avesha/ritesh/paper-repro
PY=.venvr/bin/python
DATA=data/ADFTD/processed
OUT=results/adftd
mkdir -p $OUT
SEEDS="41 42 43"
COMMON="--data-dir $DATA --batch-size 256 --pre-epochs 100 --ft-epochs 100 --patience 10 --seeds $SEEDS"
run () { # name, extra args
local name=$1; shift
if [ -s "$OUT/$name.json" ]; then echo "skip $name"; return; fi
echo "=== $name ==="
$PY tsfp/train.py $COMMON "$@" --out "$OUT/$name.json" 2>&1 | tail -4
}
run tsfp_scratch --model tsfp --mode scratch
run tsfp_rec --model tsfp --mode pretrain_ft --use-div 0
run tsfp_rec_div --model tsfp --mode pretrain_ft --use-div 1
run timae_scratch --model timae --mode scratch
run timae_pretrain --model timae --mode pretrain_ft
run simmtm_scratch --model simmtm --mode scratch
run simmtm_pretrain --model simmtm --mode pretrain_ft
echo "ALL DONE"
|