KDM

Complex Event Schema Induction with Knowledge-Enriched Diffusion Model

To train and test the diffusion model in the ESG module:

train

python ESG_single.py --dataset=suicide_ied --seed=1

test

python ESG_single.py --dataset=suicide_ied --seed=1 --predictor=True

To train and test the conditional diffusion model in the ESG module:

train

python ESG.py --seed=1

test

python ESG.py --seed=1 --predictor=True

To train and test the ERP module:

train

python ERP.py --seed=1

test

python ERP.py --seed=1 --predict=True