pardonmyai-tiny / README.md
tarekziade's picture
Update README.md
3c45170 verified
metadata
license: apache-2.0
widget:
  - text: These are nice flowers
    example_tile: not offensive
  - text: What the hell
    example_tile: offensive
  - text: You really suck, dude
    example_tile: offensive2
  - text: How to put screw thread in furniture?
    example_title: not offensive but with the word screw
base_model: huawei-noah/TinyBERT_General_4L_312D
datasets:
  - tarekziade/profanity
library_name: transformers

Fine-tuned model that detects profanity in text.

This is the "tiny" version, see https://huggingface.co/tarekziade/pardonmyai

Evaluation:

  • Accuracy: 0.9633
  • Precision: 0.9378
  • Recall: 0.8746
  • F1 Score: 0.9051
  • AUC-ROC: 0.9892