Update README.md
Browse files
README.md
CHANGED
|
@@ -6,6 +6,10 @@ tags:
|
|
| 6 |
licenses:
|
| 7 |
- cc-by-nc-sa
|
| 8 |
license: openrail++
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
---
|
| 10 |
|
| 11 |
## Toxicity Classification Model
|
|
@@ -29,7 +33,7 @@ predicted_label = torch.sigmoid(output.logits).argmax().item()
|
|
| 29 |
|
| 30 |
## Citation
|
| 31 |
|
| 32 |
-
|
| 33 |
|
| 34 |
```
|
| 35 |
@inproceedings{logacheva-etal-2022-paradetox,
|
|
|
|
| 6 |
licenses:
|
| 7 |
- cc-by-nc-sa
|
| 8 |
license: openrail++
|
| 9 |
+
base_model:
|
| 10 |
+
- FacebookAI/roberta-large
|
| 11 |
+
datasets:
|
| 12 |
+
- google/jigsaw_toxicity_pred
|
| 13 |
---
|
| 14 |
|
| 15 |
## Toxicity Classification Model
|
|
|
|
| 33 |
|
| 34 |
## Citation
|
| 35 |
|
| 36 |
+
To acknowledge our work, please, use the corresponding citation:
|
| 37 |
|
| 38 |
```
|
| 39 |
@inproceedings{logacheva-etal-2022-paradetox,
|