#!/bin/bash
#SBATCH --job-name=EHD_ENS
#SBATCH --nodes=1
#SBATCH --exclusive
#SBATCH --time=02:00:00
#SBATCH --output=/home/tanpv/Progs/claude/HW/logs/ens_%j.out
#SBATCH --error=/home/tanpv/Progs/claude/HW/logs/ens_%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}
cd "$ROOT"
export OMP_NUM_THREADS=20 MKL_NUM_THREADS=20
TAGS=${TAGS:-"- _lgbmF3 _mlp"}
TUNE=${TUNE:-}
$PY src/ensemble.py --tags $TAGS --out-tag _ens $TUNE
$PY src/train_national.py finalize --model ensemble --fset F3 --tag _ens --neg-keep 0.15
$PY src/make_ai_html.py
echo "=== done ==="