This is the code base for ANLP HUJI course exercise 1, fine tuning pretrained models to perform sentiment analysis on the SST2 dataset.
pip install -r requirements.txt
Run:
python ex1.py <number of seeds> <number of training samples> <number of validation samples> <number of prediction samples>
Generated files are res.txt, showing model's performance, training time and prediction time, and prediction.txt, containing prediction results for all test samples.