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

2026-06-22 17:31:32 +09:00
commit 8b7f70d650
26 changed files with 3445 additions and 0 deletions
@@ -0,0 +1,25 @@
+#!/bin/bash
+set -e
+cd "/Users/kyo/Documents/AY/논문/5. WCL-DRL"
+EMB="bert_agnews_8000.pt"
+PY="/opt/anaconda3/bin/python"
+# seed=0 already covered by results_drl_long / results_drl_long_joint
+for S in 42 123; do
+    echo ">>> Joint U=4 200ep seed=$S"
+    $PY Code/drl_mask_policy.py \
+        --mode joint --embed-file $EMB \
+        --users 4 --mux-factor 4 \
+        --epochs 200 --steps-per-epoch 200 \
+        --seed $S \
+        --save-dir results_sweeps/joint_U4_s${S}_200ep \
+        2>&1 | tail -2
+    echo ">>> DRL U=4 200ep seed=$S"
+    $PY Code/drl_mask_policy.py \
+        --mode drl --embed-file $EMB \
+        --users 4 --mux-factor 4 \
+        --epochs 200 --steps-per-epoch 200 \
+        --seed $S \
+        --save-dir results_sweeps/drl_U4_s${S}_200ep \
+        2>&1 | tail -2
+done
+echo "=== 3-seed 200ep sweep done ==="