From f99898c124dcef4e8dc80b42e7588420e2580813 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Wed, 5 Apr 2023 10:46:35 +0200
Subject: [PATCH] wip
---
convlab/policy/emoTUS/analysis.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/convlab/policy/emoTUS/analysis.py b/convlab/policy/emoTUS/analysis.py
index 829ed134..4862a2e2 100644
--- a/convlab/policy/emoTUS/analysis.py
+++ b/convlab/policy/emoTUS/analysis.py
@@ -41,6 +41,12 @@ def advance(conversation):
def get_turn_emotion(conversation):
+ """ Get the emotion of each turn in the conversation
+ Args:
+ conversation (list): a list of dialog
+ Returns:
+ turn_emotion (list): a list of emotion of each turn
+ """
turn_info = {"all": {},
"Complete": {}, "Not Complete": {},
"Success": {}, "Not Success": {},
--
GitLab