waltbai/RePredictor

Rich Event Modeling for Script Event Prediction

RePredictor

Experiment code for:

Long Bai, Saiping Guan, Zixuan Li, Jiafeng Guo, Xiaolong Jin, Xueqi Cheng. "Rich Event Modeling for Script Event Prediction", AAAI 2023

Install

Install this package via:

pip install .

Code for generating MCNC-rich can be found here.

Preprocess

python scripts/preprocess.py --config <ConfigFile>

Train

python scripts/train.py --config <ConfigFile> --device <Device>

Test

python scripts/test.py --config <ConfigFile> --device <Device>