Initial release: code for WCL2026-1544 (context-aware embedding masking via DRL)

This commit is contained in:
Ki-Ho Lee
2026-06-22 17:31:32 +09:00
commit 8b7f70d650
26 changed files with 3445 additions and 0 deletions
+15
View File
@@ -0,0 +1,15 @@
#!/bin/bash
set -e
cd "/Users/kyo/Documents/AY/논문/5. WCL-DRL"
EMB="bert_agnews_8000.pt"
PY="/opt/anaconda3/bin/python"
for U in 2 3 4 5 6; do
echo ">>> DRL seed=123 U=$U"
$PY Code/drl_mask_policy.py \
--mode drl --embed-file $EMB \
--users $U --mux-factor 4 \
--epochs 60 --steps-per-epoch 150 --seed 123 \
--save-dir results_sweeps/drl_U${U}_s123 \
2>&1 | tail -3
done
echo "=== seed=123 done ==="