Update json2text.py

elgeish · Jun 26, 2019 · 2baa793 · 2baa793
1 parent e58070d
commit 2baa793
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/utils/json2text.py b/utils/json2text.py
@@ -51,7 +51,8 @@ def get_parser():
 
     for x in j['utts']:
         seq = [char_list[int(i)] for i in j['utts'][x]['output'][0]['rec_tokenid'].split()]
-        h.write(" ".join(seq).replace('<eos>', '') + '\n')
+        h.write(x + " " + " ".join(seq).replace('<eos>', '') + '\n')
 
-        seq = [char_list[int(i)] for i in j['utts'][x]['output'][0]['tokenid'].split()]
-        r.write(" ".join(seq).replace('<eos>', '') + '\n')
+        if 'tokenid' in j['utts'][x]['output'][0].keys():
+            seq = [char_list[int(i)] for i in j['utts'][x]['output'][0]['tokenid'].split()]
+            r.write(x + " " + " ".join(seq).replace('<eos>', '') + '\n')