WSDM2022 "A Simple but Effective Bidirectional Extraction Framework for Relational Triple Extraction"
The main requirements are:
- python 3.6
- torch 1.4.0
- tqdm
- transformers == 2.8.0
- Train and select the model
python run.py --dataset=WebNLG --train=train --batch_size=6
python run.py --dataset=WebNLG_star --train=train --batch_size=6
python run.py --dataset=NYT --train=train --batch_size=18
python run.py --dataset=NYT_star --train=train --batch_size=18
python run.py --dataset=NYT10 --train=train --batch_size=18
python run.py --dataset=NYT11 --train=train --batch_size=18
- Evaluate on the test set
python run.py --dataset=WebNLG --train=test
python run.py --dataset=WebNLG_star --train=test
python run.py --dataset=NYT --train=test
python run.py --dataset=NYT_star --train=test
python run.py --dataset=NYT10 --train=test
python run.py --dataset=NYT11 --train=test
