language: | |
- en | |
license: mit | |
datasets: | |
- cardiffnlp/x_sensitive | |
metrics: | |
- f1 | |
widget: | |
- text: Call me today to earn some money mofos! | |
pipeline_tag: text-classification | |
# twitter-roberta-base-sensitive-binary | |
This is a RoBERTa-base model trained on 154M tweets until the end of December 2022 and finetuned for detecting sensitive content (binary classification) on the [_X-Sensitive_](https://huggingface.co/datasets/cardiffnlp/x_sensitive) dataset. | |
The original Twitter-based RoBERTa model can be found [here](https://huggingface.co/cardiffnlp/twitter-roberta-base-2022-154m). | |
## Labels | |
``` | |
"id2label": { | |
"0": "non-sensitive", | |
"1": "sensitive" | |
} | |
``` | |
## Full classification example | |
```python | |
from transformers import pipeline | |
pipe = pipeline(model='cardiffnlp/twitter-roberta-base-sensitive-binary') | |
text = "Call me today to earn some money mofos!" | |
pipe(text) | |
``` | |
Output: | |
``` | |
[{'label': 'sensitive', 'score': 0.999821126461029}] | |
``` | |
## BibTeX entry and citation info | |
TBA |