Training scripts for EUREKA: EUphemism Recognition Enhanced Through KNN-based Methods and Augmentation
To train, call python train.py
, which will create a saved model for every checkpoint at the specified output path. See train.py
for specific arguments.
To evaluate, call python evaluate.py --model_path (checkpoint-path)/pytorch_model.bin
.