The implementation of NAACL 2021 paper "Event Representation with Sequential, Semi-Supervised Discrete Variables".
This Pytorch code implements the model and reproduces the results from the paper.
conda create --name ssdvae
conda install pytorch==0.4.1 cuda92 -c pytorch
pip install torchtext==0.2.3
conda install pandas scikit-learn
conda install -c conda-forge pytablewriter
Wikipedia Inverse Narrative Cloze
./train.sh $obsv_prob $exp_num $seed
{'valid','test'}
./test_ppx.sh $obsv_prob $exp_num $seed $data_mode
./wiki_inv_narr.sh $obsv_prob $exp_num $seed $data_mode
./nyt_inv_narr.sh $obsv_prob $exp_num $seed $data_mode
Some parts of the code were inspired by HAQAE implementations.