From 00271b5975d2687c25c5ddc19e4b4f29e56e3e91 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Thu, 12 Jan 2023 01:41:24 +0100
Subject: [PATCH] wip

---
 convlab/policy/emoTUS/evaluate.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py
index 04baca8d..011ec365 100644
--- a/convlab/policy/emoTUS/evaluate.py
+++ b/convlab/policy/emoTUS/evaluate.py
@@ -38,6 +38,7 @@ def arg_parser():
                         help="do golden nlg generation")
     parser.add_argument("--no-neutral", action="store_true",
                         help="skip neutral emotion")
+    parser.add_argument("--use-sentiment", action="store_true")
     return parser.parse_args()
 
 
@@ -315,7 +316,8 @@ def main():
     eval = Evaluator(args.model_checkpoint,
                      args.dataset,
                      args.model_weight,
-                     args.only_action)
+                     args.only_action,
+                     args.use_sentiment)
     print("model checkpoint", args.model_checkpoint)
     print("generated_file", args.generated_file)
     print("input_file", args.input_file)
-- 
GitLab