(code)
| 4 | |
| 5 | |
| 6 | def postprocess(code): |
| 7 | code = code.replace("<NUM_LIT>", "0").replace("<STR_LIT>", "").replace("<CHAR_LIT>", "") |
| 8 | pattern = re.compile(r"<(STR|NUM|CHAR)_LIT:(.*?)>", re.S) |
| 9 | lits = re.findall(pattern, code) |
| 10 | for lit in lits: |
| 11 | code = code.replace(f"<{lit[0]}_LIT:{lit[1]}>", lit[1]) |
| 12 | return code |
| 13 | |
| 14 | |
| 15 | def eval(predicted_sequences, ground_truths): |