CLUSTERING_SCRIPT="src/aurora/eval/classification/kmeans/clustering.py"
OUTPUT_DIR="src/aurora/eval/classification/kmeans/clustering4"

for SEED in 42 100 200; do
    for CONCAT_EMBS in 0 1; do
        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_chronos_t5_tiny_trn_val_tst_ctx200_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_chronos_bolt_tiny_trn_val_tst_ctx200_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_chronos_bolt_trn_val_tst_ctx200_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_chronos_t5_trn_val_tst_ctx200_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw_embs_moiral_small_trn_val_tst_ctx200_pdt64_psz16_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw_embs_moiral_trn_val_tst_ctx200_pdt64_psz16_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_hand_crafted_trn_val_tst_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS \
        #     --hand-crafted 1 &

        # python $CLUSTERING_SCRIPT \
        #     --dataset-dir "/embs/embs_random_trn_val_tst_bandgr" \
        #     --save-dir "$OUTPUT_DIR" \
        #     --mode test \
        #     --perplexity 30 \
        #     --random-state $SEED \
        #     --concat-embs $CONCAT_EMBS \
        #     --hand-crafted 1 &

        python $CLUSTERING_SCRIPT \
            --dataset-dir "hf_csdr1_multiband_raw4_embeddings_astromer_1_subclass_pad_correct" \
            --save-dir "$OUTPUT_DIR" \
            --mode test \
            --perplexity 30 \
            --random-state $SEED \
            --concat-embs $CONCAT_EMBS &

        # wait
    done
done


# python clustering_tmp.py --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_chronos_bolt_trn_val_tst_ctx200_bandgr" --save-dir "outputs/clustering_tmp" --mode test --perplexity 30 --random-state 42 --concat-embs 1


# python "src/aurora/eval/classification/kmeans/clustering.py" --dataset-dir "embs/csdr1_raw4_catflags_filtered_embs_hand_crafted_trn_val_tst_bandgr" --save-dir "src/aurora/eval/classification/kmeans/clustering4" --mode test --perplexity 30 --random-state 42 --concat-embs 0