CafeBERT Fine-tuned for Toxic Classification

This model is a fine-tuned version of CafeBERT on a toxic comment classification task.

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("UngLong/cafebert-classification-ft")
model = AutoModelForSequenceClassification.from_pretrained("UngLong/cafebert-classification-ft")
Downloads last month
53
Safetensors
Model size
560M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for UngLong/cafebert-classification-ft

Base model

uitnlp/CafeBERT
Finetuned
(3)
this model

Dataset used to train UngLong/cafebert-classification-ft