antypasd's picture
Upload RobertaForSequenceClassification
669aae2 verified
|
raw
history blame
1.01 kB
---
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