Description
This repository contains the code for "Exploiting Cloze Questions for Few-Shot Text Classification and Natural Language Inference" - timoschick/pet
Summary
- 📑 Contents 🔧 Setup All requirements for PET can be found in requirements.txt.
- $DATA_DIR is the directory containing the train and test files (check tasks.py to see how these files should be named and formatted for each task).
- You can also define multiple verbalizations for a single label.
- If you want to use verbalizers that correspond to multiple tokens (as described here), you need to define a custom TaskHelper and add it to the TASK_HELPERS dictionary in pet/tasks.py.