Skip to content

dipta007/SHEM

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Semantically-informed Hierarchical Event Modeling

Python Version MIT License GitHub Top Language

This repository is the official implementation of Semantically-informed Hierarchical Event Modeling (Published in *SEM 2023).

Main Figure

Getting Started

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

Conda Environment

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

Data:

conda activate shem
pip install gdown
gdown https://drive.google.com/drive/u/1/folders/1s16s33Fwgt1MNk7_YO7iJO32gW2TZvrv -O ./data --folder

Usage

Training:

python train.py +experiment=naacl ++exp_num=$exp_num ++seed=$SEED ++debug=False

Evaluation:

test_type:

{'sim-hard', 'sim-ext', 'sim-ext'}

similarity evaluation:

python test.py --exp_name=naacl_$exp_num --test_type=$test_type

About

Semantically-informed Hierarchical Event Modeling

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors