Downloads · 30 days
4
15% of all-time downloads
Jensvollends/hatebert-finetuned
hatebert-finetuned is a machine learning model from Jensvollends. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-sa-4.0.
This model is a fine-tuned version of GroNLP/hateBERT on the Jigsaw Toxic Comment Classification Challenge dataset.
Downloads · 30 days
4
15% of all-time downloads
All-time downloads
26
Public
Parameters
109M
3.9 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt2.6 GB · 60%
From the Hugging Face model README
This model is a fine-tuned version of GroNLP/hateBERT on the Jigsaw Toxic Comment Classification Challenge dataset.
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("Jensvollends/hatebert-finetuned")
tokenizer = AutoTokenizer.from_pretrained("Jensvollends/hatebert-finetuned")
inputs = tokenizer("You are a bad person", return_tensors="pt")
outputs = model(**inputs)