Downloads · 30 days
10
6% of all-time downloads
distilbert-toxic-comment-classifier
longluu/distilbert-toxic-comment-classifier
distilbert-toxic-comment-classifier is a text classification model from longluu. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
Use a pretrained DistilBert to train a classifier on the Toxic Comment dataset https://www.kaggle.com/c/jigsaw-toxic-comment-classification-challenge. The goal is to classify whether a comment is toxic or not. Note th…
All-time downloads
176
Public
Parameters
67M
268 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- distilbert
- Access
- Public
- Created
- Jan 21, 2024
- Updated
- Jan 23, 2024
- SHA
- f3af0bb5
Try a prompt
Model card
From the Hugging Face model README
Toxic post classification using DistilBert
Use a pretrained DistilBert to train a classifier on the Toxic Comment dataset https://www.kaggle.com/c/jigsaw-toxic-comment-classification-challenge. The goal is to classify whether a comment is toxic or not. Note that the labels from the original datasets are more fine-grained (i.e. different types of toxicity). The model here obatains a test accuracy of 95% on a balanced split.
- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- mit
- Languages
- en
- Created
- Jan 21, 2024
- Updated
- Jan 23, 2024