--- a +++ b/shepherd/run_patients_like_me.sh @@ -0,0 +1,18 @@ +# Bash script to train SHEPHERD for patients-like-me + +# Command to run this bash script: +# bash run_patients_like_me.sh + +# Command to run with the best hyperparameters from the paper +python train.py \ + --edgelist KG_edgelist_mask.txt \ + --node_map KG_node_map.txt \ + --patient_data disease_simulated \ + --run_type patients_like_me \ + --saved_node_embeddings_path checkpoints/pretrain.ckpt \ + --sparse_sample 300 \ + --lr 5e-05 \ + --upsample_cand 1 \ + --neighbor_sampler_size 15 \ + --lmbda 0.7 \ + --kappa 0.09000000000000001 \ No newline at end of file