"PTR: Prompt Tuning with Rules for Text Classification"
- Fixed Version for KLUE-RE
If you use the code, please cite the following paper:
@article{han2021ptr,
title={PTR: Prompt Tuning with Rules for Text Classification},
author={Han, Xu and Zhao, Weilin and Ding, Ning and Liu, Zhiyuan and Sun, Maosong},
journal={arXiv preprint arXiv:2105.11259},
year={2021}
}
The model is implemented using PyTorch. The versions of packages used are shown below.
-
numpy>=1.18.0
-
scikit-learn>=0.22.1
-
scipy>=1.4.1
-
torch>=1.3.0
-
tqdm>=4.41.1
-
transformers>=4.0.0
We provide prompts used in our experiments.