我們的 EMNLP 2021 論文的官方 PyTorch 實作和 EmoCause 評估集?
金賢宇、金秉昌和金健熙。換位思考和語用學,以產生針對情感原因的同理心反應。 EMNLP ,2021 [論文]
如果您使用此儲存庫中的資料作為任何已發表研究的一部分,我們要求您引用以下論文:
@inproceedings { Kim:2021:empathy ,
title = { Perspective-taking and Pragmatics for Generating Empathetic Responses Focused on Emotion Causes } ,
author = { Kim, Hyunwoo and Kim, Byeongchang and Kim, Gunhee } ,
booktitle = { EMNLP } ,
year = 2021
}
我們的程式碼建構在 ParlAI 框架上。我們建議您建立一個 conda 環境,如下所示
conda env create -f environment.yml
並激活它
conda activate focused-empathy
python -m spacy download en
EmoCause是來自 EmpatheticDialogues 有效和測試集的情感情境中帶註釋的情感原因詞的資料集。目標是透過僅在沒有單字級標籤的句子級情緒標籤上進行訓練來識別句子中的情緒原因詞(即弱監督情感原因識別)。 EmoCause的基礎是人類無法透過詞級原因標籤的監督學習來辨識情緒的原因。因此,我們不提供訓練集。
您可以[此處]下載EmoCause評估集。
請注意,當您執行下面的實驗命令時,將自動下載資料集。
#情感 | 標籤類型 | #標籤/話語 | #發聲 | |
---|---|---|---|---|
情感事業 | 32 | 單字 | 2.3 | 4.6K |
{
"original_situation": the original situations in the EmpatheticDialogues,
"tokenized_situation": tokenized situation utterances using spacy,
"emotion": emotion labels,
"conv_id": id for each corresponding conversation in EmpatheticDialogues,
"annotation": list of tuples: (emotion cause word, index),
"labels": list of strings containing the emotion cause words
}
執行以下命令時,將自動下載所有對應的模型。
我們也提供手冊下載連結:[GEE] [Finetuned Blender]
您可以在 EmoCause 評估集上評估我們提出的生成情緒估計器 (GEE)。
python eval_emocause.py --model agents.gee_agent:GeeCauseInferenceAgent --fp16 False
您可以評估我們在微調的 Blender 上產生集中同理心回應的方法(不熟悉 Blender?請參閱此處!)。
python eval_empatheticdialogues.py --model agents.empathetic_gee_blender:EmpatheticBlenderAgent --model_file data/models/finetuned_blender90m/model --fp16 False --empathy-score False
新增--alpha 0
標誌將執行 Blender,無需使用任何語法。您也可以透過新增--distractor-type random
來嘗試隨機幹擾器 (Plain S1)。
若要同時測量解釋和探索分數,請將--empathy-score
設定為True
。它將自動下載在 EmpatheticDialogues 上微調的 RoBERTa 模型。有關同理心得分的更多詳細信息,請訪問原始存儲庫。
我們感謝匿名審稿人對這項工作的有益評論。
這項研究得到了三星電子三星研究資助中心的支持,項目編號為 SRFCIT210101。計算資源和人類研究得到韓國國家研究基金會 (NRF) 腦研究計劃 (2017M3C7A1047860) 的支持。
請透過 hyunw.kim 聯繫 Hyunwoo Kim,地址為 vl dot snu dot ac dot kr。
該存儲庫已獲得 MIT 許可。有關詳細信息,請參閱許可證文件。