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