bdotloh/empathetic-dialogues-contexts
Viewer • Updated • 24.5k • 608 • 25
How to use bdotloh/distilbert-base-uncased-empathetic-dialogues-context with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="bdotloh/distilbert-base-uncased-empathetic-dialogues-context") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("bdotloh/distilbert-base-uncased-empathetic-dialogues-context")
model = AutoModelForSequenceClassification.from_pretrained("bdotloh/distilbert-base-uncased-empathetic-dialogues-context")This is a distilbert-base-uncased model fine-tuned for the purpose of classifying (emotional) contexts in the Empathetic Dialogues dataset.
EmpatheticDialogues:
see dataset
Top 1 accuracy: 53.4 Top 5 accuracy: 86.1