zhuyiYZU / PLST

Prompt-Learing for Short Text Classification
10 stars 4 forks source link

This is the repository for the "Prompt-Learing for Short Text Classification".

First you need to install OpenPrompt. see https://github.com/thunlp/OpenPrompt

Then you need to use the methods to filter tag words,In this process, you need to download the corresponding vocabulary. These words need to be put scripts/TextClassification,and modify the corresponding position.

Also, you can put your own dataset in datasets/TextClassification.

example shell scripts:

python main.py --verbalizer cpt