tarekziade/pardonmyai

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