sgc.sh

dataset=$1

for lm_model in 'Salesforce/SFR-Embedding-Mistral' # 'McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp' 'Alibaba-NLP/gte-Qwen1.5-7B-instruct'
do
    results_file="./results2/$dataset/sgc-undirected-$lm_model.jsonl"
    mkdir -p `dirname $results_file`

    for seed in "3" "12" "42" "69"
    do
    echo "---------------------------------------------------------------------------------"
    echo "Running pipeline for dataset '$dataset' seed '$seed' LM '$lm_model'"
    echo "---------------------------------------------------------------------------------"

    python -m core.trainGNN \
        dataset ${dataset} \
        seed ${seed} \
        lm.model.name ${lm_model} \
        gnn.model.name SimpleGCN \
        gnn.diffusion.k 2 \
        gnn.diffusion.tag "-undirected" \
        gnn.train.wd 0.000005 \
        gnn.train.early_stop 0 \
        results_file ${results_file}

    done
done