distilbert-base-uncased-finetuned-sst-2-english-openvino
echarlaix
Clasificación de texto
Este modelo es un punto de control afinado de DistilBERT-base-uncased, afinado en SST-2. Este modelo alcanza una precisión de 91.3 en el conjunto de desarrollo.
Como usar
Puedes usar este modelo con Transformers pipeline.
from transformers import AutoTokenizer, pipeline
from optimum.intel.openvino import OVModelForSequenceClassification
model_id = "echarlaix/distilbert-base-uncased-finetuned-sst-2-english-openvino"
model = OVModelForSequenceClassification.from_pretrained(model_id)
tokenizer = AutoTokenizer.from_pretrained(model_id)
cls_pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
text = "He's a dreadful magician."
outputs = cls_pipe(text)
Funcionalidades
- Clasificación de texto
- Transformers
- OpenVINO
- sst2
- glue
- Inferencia
Casos de uso
- Clasificación de texto