MiniLMv2-toxic-jigsaw
minuva
Clasificación de texto
Este modelo es una versión ajustada finamente de MiniLMv2-L6-H384 en el conjunto de datos de la primera competencia Kaggle de Jigsaw utilizando unitary/toxic-bert como modelo maestro. La versión cuantizada en formato ONNX se puede encontrar aquí. El modelo con solo dos etiquetas (toxicidad y toxicidad severa) está aquí.
Como usar
from transformers import pipeline
pipe = pipeline(model='minuva/MiniLMv2-toxic-jigsaw', task='text-classification')
pipe("This is pure trash")
# [{'label': 'toxic', 'score': 0.9383478164672852}]
Funcionalidades
- Clasificación de texto
- Transformers
- ONNX
- Safetensors
- Clasificación de toxicidad
- Inferencia
- Entrenado usando unitary/toxic-bert
Casos de uso
- Clasificación de lenguaje ofensivo
- Detección de discurso de odio