metadata
license: apache-2.0
datasets:
- UngLong/ViSIR
language:
- vi
metrics:
- f1
- recall
- precision
base_model:
- uitnlp/CafeBERT
pipeline_tag: text-classification
tags:
- socialmedia
- toxiccomment
- classification_toxic_comment
- transformer
library_name: transformers
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")