Code for text augmentation method leveraging large-scale language models

Code for our paper GPT3Mix and conducting classification experiments using GPT-3 prompt-based data augmentation.

Getting Started

The main depedencies can be installed via pip install -r requirements.txt.

The main code is run through main.py. Check out --help for full list of commands.

The code will automatically use the first GPU device, if detected.

A typical command to run BERT-base 10 times on the 1% subsample set of the SST-2 dataset and computing the average
of all run is as follows.