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