Emotion Entailment
You can test the model at Emotion Entailment | SGNLP-Demo.<br /> If you want to find out more information, please contact us at sg-nlp@aisingapore.org.
Table of Contents
Model Details
Model Name: Emotion Entailment
- Description: This is an emotion entailment model based on RoBERTa base which recognises the cause behind emotions in conversations. Given 4 sets of inputs: target utterance, target utterance's emotion, evidence utterance and conversational history, it returns the probability of the evidence utterance causing the emotion specified in the target utterance.
- Paper: Recognizing emotion cause in conversations. arXiv preprint arXiv:2012.11820., Dec 2020.
- Author(s): Poria, S., Majumder, N., Hazarika, D., Ghosal, D., Bhardwaj, R., Jian, S.Y.B., Hong, P., Ghosh, R., Roy, A., Chhaya, N., Gelbukh, A. and Mihalcea, R. (2020).
- URL: https://arxiv.org/abs/2012.11820
How to Get Started With the Model
Install Python package
SGnlp is an initiative by AI Singapore's NLP Hub. They aim to bridge the gap between research and industry, promote translational research, and encourage adoption of NLP techniques in the industry. <br><br> Various NLP models, other than aspect sentiment analysis are available in the python package. You can try them out at SGNLP-Demo | SGNLP-Github.
pip install sgnlp
Examples
For more full code (such as Emotion Entailment), please refer to this SGNLP-Docs. <br> Alternatively, you can also try out the Emotion Entailment | SGNLP-Demo for Emotion Entailment.
Example of Emotion Entailment (for happiness):
from sgnlp.models.emotion_entailment import (
RecconEmotionEntailmentConfig,
RecconEmotionEntailmentTokenizer,
RecconEmotionEntailmentModel,
RecconEmotionEntailmentPreprocessor,
RecconEmotionEntailmentPostprocessor,
)
# Load model
config = RecconEmotionEntailmentConfig.from_pretrained(
"https://storage.googleapis.com/sgnlp-models/models/reccon_emotion_entailment/config.json"
)
tokenizer = RecconEmotionEntailmentTokenizer.from_pretrained("roberta-base")
model = RecconEmotionEntailmentModel.from_pretrained(
"https://storage.googleapis.com/sgnlp-models/models/reccon_emotion_entailment/pytorch_model.bin",
config=config,
)
preprocessor = RecconEmotionEntailmentPreprocessor(tokenizer)
postprocessor = RecconEmotionEntailmentPostprocessor()
# Model predict
input_batch = {
"emotion": ["happiness", "happiness", "happiness", "happiness"],
"target_utterance": [
"Thank you very much .",
"Thank you very much .",
"Thank you very much .",
"Thank you very much .",
],
"evidence_utterance": [
"It's very thoughtful of you to invite me to your wedding .",
"How can I forget my old friend ?",
"My best wishes to you and the bride !",
"Thank you very much .",
],
"conversation_history": [
"It's very thoughtful of you to invite me to your wedding . How can I forget my old friend ? My best wishes to you and the bride ! Thank you very much .",
"It's very thoughtful of you to invite me to your wedding . How can I forget my old friend ? My best wishes to you and the bride ! Thank you very much .",
"It's very thoughtful of you to invite me to your wedding . How can I forget my old friend ? My best wishes to you and the bride ! Thank you very much .",
"It's very thoughtful of you to invite me to your wedding . How can I forget my old friend ? My best wishes to you and the bride ! Thank you very much .",
],
}
tensor_dict = preprocessor(input_batch)
raw_output = model(**tensor_dict)
output = postprocessor(raw_output)
Training
The train and evaluation datasets were derived from the RECCON dataset. The full dataset can be downloaded from the author's github repository.
Training Results
- Training Time: ~3 hours for 12 epochs on a single V100 GPU.
Model Parameters
- Model Weights: link
- Model Config: link
- Model Inputs: Target utterance, emotion in target utterance, evidence utterance and conversational history.
- Model Outputs: Probability score of whether evidence utterance caused target utterance to exhibit the emotion specified.
- Model Size: ~477MB
- Model Inference Info: ~ 2 sec on Intel(R) Core(TM) i7-8750H CPU @ 2.20GHz.
- Usage Scenarios: Recognizing emotion cause for phone support satisfaction.
Other Information
- Original Code: link
