pip install -r requirements.txt- For AGNews and DBpedia, dataset will be automatically downloaded during the training.
- For SST-2, please download the dataset from GLUE Benchmark and put the files into
./data/sst2/.
We use json file for the configuration. Before running the code, please check hyperparameters.json and select proper parameters.
Then just simply run:
python -m associated_learning.main