#!/bin/bash
#SBATCH --account=xxx
#SBATCH --job-name=rf_classification
#SBATCH --nodes=1
#SBATCH --output=xxx
#SBATCH --error=xxx
#SBATCH --time=1:00:00
#SBATCH --partition=gengpu
#SBATCH --gres=gpu:h100:1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=16
#SBATCH --open-mode=append
#SBATCH --mem=60G
#SBATCH --mail-type=ALL ## you can receive e-mail alerts from SLURM when your job begins and when your job finishes (completed, failed, etc)
#SBATCH --mail-user=xxx ## your email


module purge all
module load gcc/11.2.0
eval "$(conda shell.bash hook)"
conda activate ag
cd /xxx


# python /xxx/rf_hpo.py \
#  --input-embs /xxx/csdr1_raw4_catflags_filtered_embs_hand_crafted_trn_val_tst_bandgr \
#  --standardize 1 \
#  --hand-crafted 1 \
#  --seed 42 \
#  --skip-hpo \
#  --best-params '{"max_depth": None, "min_samples_split": 10, "n_estimators": 100}' \
#  --output-dir /xxx/handcrafted_feature


# python /xxx/rf_hpo.py \
#  --input-embs /xxx/hf_csdr1_multiband_raw4_embeddings_astromer_2_gr_sampling_True \
#  --hand-crafted 0 \
#  --seed 42 \
#  --skip-hpo True\
#  --best-params '{"max_depth": 30, "min_samples_split": 10, "n_estimators": 500}' \
#  --output-dir /xxx/output/rf/astromer_2



# python /xxx/rf_hpo.py \
#  --input-embs /xxx/csdr1_raw4_catflags_filtered_embs_chronos_bolt_tiny_trn_val_tst_ctx200_bandgr \
#  --hand-crafted 0 \
#  --seed 42 \
#  --output-dir /xxx/output/rf/chronos_bolt_tiny

# python /xxx/rf_hpo.py \
#  --input-embs /xxx/csdr1_raw_embs_moiral_small_trn_val_tst_ctx200_pdt64_psz16_bandgr \
#  --hand-crafted 0 \
#  --seed 42 \
#  --output-dir /xxx/output/rf/moirai

# python /xxx/rf_hpo.py \
#  --input-embs /xxx/hf_csdr1_multiband_raw4_embeddings_astromer_1_subclass_pad_correct \
#  --hand-crafted 0 \
#  --seed 42 \
#  --output-dir /xxx/output/rf/astromer_1

# python /xxx/rf_hpo.py \
#  --input-embs /xxx/csdr1_raw4_catflags_filtered_embs_chronos_t5_tiny_trn_val_tst_ctx200_bandgr \
#  --hand-crafted 0 \
#  --seed 42 \
#  --output-dir /xxx/output/rf/chronos_tiny

python /xxx/rf_hpo.py \
 --input-embs /xxx/random \
 --hand-crafted 1 \
 --seed 42 \
 --output-dir /xxx/output/rf/random 



