This repository is the official implementation of Semantically-informed Hierarchical Event Modeling (Published in *SEM 2023).
git clone https://github.com/dipta007/SHEM
cd SHEM
git checkout event_similarity
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 hydra-core --upgrade
pip install hydra_colorlog --upgrade
pip install wandb
pip install prettytable
pip install transformers
conda install -c conda-forge spacy
conda install -c conda-forge cupy
python -m spacy download en_core_web_trf
pip install gdown
conda activate shem
pip install gdown
gdown https://drive.google.com/drive/u/1/folders/1s16s33Fwgt1MNk7_YO7iJO32gW2TZvrv -O ./data --folder
python train.py +experiment=naacl ++exp_num=$exp_num ++seed=$SEED ++debug=False
{'sim-hard', 'sim-ext', 'sim-ext'}
python test.py --exp_name=naacl_$exp_num --test_type=$test_type
