diff --git a/convlab2/nlg/sclstm/multiwoz/sc_lstm.py b/convlab2/nlg/sclstm/multiwoz/sc_lstm.py index 9171efb85ffc78d7bcd383aead8556fd3f97f1ad..26eb1dc3b19c49f19b282b6bd582fec61a7a54e5 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: