nikokks / plst

Prompt-Learing for Short Text Classification

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

see paper one arxiv : https://arxiv.org/abs/2202.11345

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

code cpt_verbalizer inspired from https://github.com/abhishek0318/Probase/tree/af6cb903c39b0fe2bf1326fdd9673e77386e087f