git clone https://github.com/dipta007/SHEM
cd SHEM
git checkout missing_grp

conda create -n shem python=3.7
conda activate shem
conda install pip
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
pip install torchtext==0.2.3
pip install -U scikit-learn
conda install -c conda-forge pytablewriter
conda install -c anaconda pandas
pip install gdown

conda activate shem
pip install gdown
mkdir saved_models
mkdir saved_configs
gdown https://drive.google.com/drive/u/1/folders/1StFmV2L5MiubGYzd5NB-weQpfun-SA6I -O ./data --folder

./train.sh $obsv_prob $exp_num $seed

./test_ppx.sh $obsv_prob $exp_num $seed $data_mode

./wiki_inv_narr.sh $obsv_prob $exp_num $seed $data_mode

./test_mppx.sh $obsv_prob $exp_num $seed $data_mode