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")