export CUDA_VISIBLE_DEVICES="0"

# SET THE CORRECT MODEL_CHECKPOINTS TO MATCH EACH CONFIGURATION

# No HRR
python3 BERTHA_Finetune_Mortality_Prediction.py --use_random_embeddings \
    --positive_class_weight=11 --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"

# HRR
python3 BERTHA_Finetuning_Mortality_Prediction.py --use_vsa --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --positive_class_weight=11 --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"

# HRR Cat
python3 BERTHA_Finetuning_Mortality_Prediction.py --use_vsa --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --positive_class_weight=11 --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"

# HRR Add
python3 BERTHA_Finetuning_Mortality_Prediction.py --use_vsa --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --positive_class_weight=11 --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"
