Fine-tuned model that detects profanity in text.
Inspired from https://victorzhou.com/blog/better-profanity-detection-with-scikit-learn/
The model was trained with the dataset from that project.
Usage example with Python:
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="tarekziade/pardonmyai")
print(classifier("These are beautiful flowers"))
Usage example with Transformers.js:
import { pipeline } from '@xenova/transformers';
let pipe = await pipeline('sentiment-analysis', model='tarekziade/pardonmyai');
let out = await pipe('These are beautiful flowers');
Source code and data: https://github.com/tarekziade/pardonmyai
metrics:
- Accuracy: 0.9748
- Precision: 0.9331
- Recall: 0.9416
- F1 Score: 0.9373
- AUC-ROC: 0.9955
There's a tiny version available: https://huggingface.co/tarekziade/pardonmyai-tiny