Downloads · 30 days
16
0% of all-time downloads
toxic-detector-distilroberta
jpcorb20/toxic-detector-distilroberta
toxic-detector-distilroberta is a text classification model from jpcorb20. Use it when you need a label for a piece of text. It is set up for transformers.
All-time downloads
13.2K
Public
Repo size
992 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin331 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- roberta
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- May 20, 2021
- SHA
- 88d1b244
Try a prompt
Model card
From the Hugging Face model README
Distilroberta for toxic comment detection
See my GitHub repo toxic-comment-server
The model was trained from DistilRoberta on Kaggle Toxic Comments with the BCEWithLogits loss for Multi-Label prediction. Thus, please use the sigmoid activation on the logits (not made to use the softmax output, e.g. like the HF widget).
Evaluation
F1 scores:
toxic: 0.72
severe_toxic: 0.38
obscene: 0.72
threat: 0.52
insult: 0.69
identity_hate: 0.60
Macro-F1: 0.61
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Created
- Mar 2, 2022
- Updated
- May 20, 2021