bdotloh/empathetic-dialogues-contexts
Viewer • Updated • 24.5k • 714 • 25
How to use bdotloh/distilbert-base-uncased-go-emotion-empathetic-dialogues-context-v2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="bdotloh/distilbert-base-uncased-go-emotion-empathetic-dialogues-context-v2") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("bdotloh/distilbert-base-uncased-go-emotion-empathetic-dialogues-context-v2")
model = AutoModelForSequenceClassification.from_pretrained("bdotloh/distilbert-base-uncased-go-emotion-empathetic-dialogues-context-v2", device_map="auto")We performed transfer learning experiments on a distilbert-base-uncased model fine-tuned on the GoEmotions dataset for the purpose of classifying (emotional) contexts in the Empathetic Dialogues dataset.
The fine-tuned distilbert-base-uncased can be found here.
GoEmotions:
EmpatheticDialogues: