Update README.md

README.md

@@ -60,11 +60,11 @@ print(text2toxicity(['я люблю нигеров', 'я люблю африка
 
 ## Training
 
-The model has been trained on the joint dataset of [OK ML Cup](https://cups.mail.ru/ru/tasks/1048) and [Babakov et.al.](https://arxiv.org/abs/2103.05345) with `Adam` optimizer, learning rate of `1e-5`, and batch size of `128` for `5` epochs. The data was not filtered in any way. A text was considered inappropriate if its inappropritateness score was higher than 0.2. The per-label ROC AUC on the dev set is:
+The model has been trained on the joint dataset of [OK ML Cup](https://cups.mail.ru/ru/tasks/1048) and [Babakov et.al.](https://arxiv.org/abs/2103.05345) with `Adam` optimizer, learning rate of `1e-5`, and batch size of `64` for `15` epochs. A text was considered inappropriate if its inappropritateness score was higher than 0.8, and appropriate - if it was lower than 0.2. The per-label ROC AUC on the dev set is:
 ```
-non-toxic  : 0.9909
-insult     : 0.9882
-obscenity  : 0.9824
-threat     : 0.9868
-dangerous  : 0.7758
+non-toxic  : 0.9937
+insult     : 0.9912
+obscenity  : 0.9881
+threat     : 0.9910
+dangerous  : 0.8295
 ```