From e34f7de3feec81d034f3fc2d19f98780063688a3 Mon Sep 17 00:00:00 2001 From: aaa123git <wandz19@mails.tsinghua.edu.cn> Date: Thu, 19 Aug 2021 18:50:57 +0800 Subject: [PATCH] fix bug. DOMAIN-INTENT-Parking-yes-VALUE is not in the template file (#214) --- convlab2/nlg/sclstm/multiwoz/sc_lstm.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/convlab2/nlg/sclstm/multiwoz/sc_lstm.py b/convlab2/nlg/sclstm/multiwoz/sc_lstm.py index 9171efb8..26eb1dc3 100755 --- a/convlab2/nlg/sclstm/multiwoz/sc_lstm.py +++ b/convlab2/nlg/sclstm/multiwoz/sc_lstm.py @@ -88,9 +88,7 @@ class SCLSTM(NLG): for pair in v: if type(pair[1]) != str: pair[1] = str(pair[1]) - if pair[0] == 'Internet' or pair[0] == 'Parking': - pair.insert(1, 'yes') - elif pair[0] == 'none': + if pair[0] == 'none': pair.insert(1, 'none') else: if pair[0] in counter: -- GitLab