From 2c5ddaec2e56a3856015afca53a0bdcb079a5203 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Mon, 6 Feb 2023 03:42:26 +0100
Subject: [PATCH] wip
---
convlab/policy/ussT5/emowoz_evaluate.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/convlab/policy/ussT5/emowoz_evaluate.py b/convlab/policy/ussT5/emowoz_evaluate.py
index 5c2586f8..644b6143 100644
--- a/convlab/policy/ussT5/emowoz_evaluate.py
+++ b/convlab/policy/ussT5/emowoz_evaluate.py
@@ -92,9 +92,7 @@ def generate_result(model_checkpoint, data, stop=-1):
do_sample=False)
output = tokenizer.batch_decode(
output, skip_special_tokens=True)[0]
- if len(output) > 1:
- print(output)
- output = "illegal"
+
if "satisfaction score" in input_text:
output = tri_convert(output)
results.append({"input_text": input_text,
--
GitLab