natas_hw
/
sempre
mirror of https://github.com/percyliang/sempre
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Better convert to postfix

This commit is contained in:
Panupong Pasupat 2016-10-26 12:05:31 -07:00
parent d200bc5eb1
commit f2ba77bd75
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tables/dump-parsers/convert-to-postfix.py
View File

@ -85,25 +85,23 @@ def convert(tree, args):
recurse(subtree[1])
answer.append(opr)
if not args.implicit_join:
answer.append('.')
answer.append('join')
else: # Join with a complex construct
assert len(subtree) == 2, str(subtree)
if args.implicit_join:
# Only allows ((reverse ...) ...)
assert subtree[0][0] == 'reverse', str(subtree)
assert len(subtree[0]) == 2, str(subtree)
recurse(subtree[1])
answer.append('!' + subtree[0][1])
else:
recurse(subtree[1])
recurse(subtree[0])
answer.append('.')
# Only allows ((reverse ...) ...)
assert subtree[0][0] == 'reverse', str(subtree)
assert len(subtree[0]) == 2, str(subtree)
recurse(subtree[1])
answer.append('!' + subtree[0][1])
if not args.implicit_join:
answer.append('join')
recurse(tree)
return answer
def process(line, args):
line = line[:-1].split('\t')
print '\t'.join([' '.join(convert(lisptree.parse(line[0]), args))] + line[1:])
line = line.rstrip('\n').split('\t')
line[args.field] = ' '.join(convert(lisptree.parse(line[args.field]), args))
print '\t'.join(line)
def main():
parser = argparse.ArgumentParser()
@ -113,6 +111,8 @@ def main():
help='Do not output "lambda reverse" for superlatives')
parser.add_argument('-t', '--collapse-type-row', action='store_true',
help='Collapse "(type row)" into a single token')
parser.add_argument('-f', '--field', type=int, default=0,
help='Field index (tab-separated; 0-based) containing the logical form')
parser.add_argument('infile')
args = parser.parse_args()