#!/bin/bash
#SBATCH --job-name=EHD_PA1
#SBATCH --nodes=1
#SBATCH --exclusive
#SBATCH --exclude=compute-1-2
#SBATCH --time=06:00:00
#SBATCH --output=/home/tanpv/Progs/claude/HW/logs/pa1_%j.out
#SBATCH --error=/home/tanpv/Progs/claude/HW/logs/pa1_%j.err
set -euo pipefail
ROOT=/home/tanpv/Progs/claude/HW
PY=/home/tanpv/.conda/envs/ehd/bin/python
export LD_LIBRARY_PATH=/home/tanpv/.conda/envs/ehd/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
export OMP_NUM_THREADS=20 MKL_NUM_THREADS=20
cd "$ROOT"
# Dac trung ngoai khong phu thuoc z-score, nhung phai dung lai sau khi
# climatology doi sang 16 nam train de moi thu nhat quan.
# F3 = chi thong tin tu mot lan chay EC.  F6 = F3 + tien su quan trac + MJO/ENSO.
$PY src/train_ai.py --per-region 3 \
    --models ec_raw logreg lgbm_clf \
    --fsets F3 F7 F8 \
    --out "$ROOT/outputs/AI_Results/model_selection_pa1b.json"
echo "=== done ==="