Rooba
/
genienlp
mirror of https://github.com/stanford-oval/genienlp.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixes

This commit is contained in:
mehrad 2019-05-29 20:57:09 -07:00
parent 4919b28627
commit 21db011ad2
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
decanlp/utils/post_process_decoded_results.py
View File

@ -25,8 +25,6 @@ Created on Aug 27, 2018
from pprint import pprint
import sys
import os
import numpy as np
import re
import argparse
from collections import defaultdict, Counter
@ -173,10 +171,10 @@ def run():
id = id.strip()
input = input.replace(r'<s>', '').strip()
gold = gold.strip().replace(r'\"', '"').replace(r'\/', '/')
if gold[0] == gold[-1] == '"':
if gold and gold[0] == gold[-1] == '"':
gold = gold[1:-1]
pred = pred.strip().replace(r'\"', '"').replace(r'\/', '/')
if pred[0] == pred[-1] == '"':
if pred and pred[0] == pred[-1] == '"':
pred = pred[1:-1]
out_raw.write(id + '\t' + input + '\t' + gold + '\t' + pred)