|
The model has been trained [here](https://git.mts.ai/ai/ml_lab/skoltech-nlp_lab/skoltech/task_oriented_TST/-/blob/main/transfer/formality_ranker_v1.ipynb) to predict for English sentences, whether they are formal or informal. |
|
|
|
Base model: `roberta-base` |
|
|
|
Datasets: [GYAFC](https://github.com/raosudha89/GYAFC-corpus) from [Rao and Tetreault, 2018](https://aclanthology.org/N18-1012) and [online formality corpus](http://www.seas.upenn.edu/~nlp/resources/formality-corpus.tgz) from [Pavlick and Tetreault, 2016](https://aclanthology.org/Q16-1005). |
|
|
|
Data augmentation: changing texts to upper or lower case; removing all punctuation, adding dot at the end of a sentence. |
|
|
|
Loss: binary classification (on GYAFC), in-batch ranking (on PT data). |
|
|