summaryrefslogtreecommitdiff
path: root/gi/pyp-topics/scripts/spans2labels.py
blob: 7cbc521ce43f1bcc3296de280b37037cb8d453ae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/python

import sys
from operator import itemgetter

if len(sys.argv) <= 2:
  print "Usage: spans2labels.py phrase_context_index [order]"
  exit(1)

order=1
if len(sys.argv) > 2:
  order = int(sys.argv[2])

phrase_context_index = {}
for line in file(sys.argv[1], 'r'):
  phrase,tail= line.split('\t')
  contexts = tail.split(" ||| ")
  assert len(contexts) % 2 == 0
  for i in range(0, len(contexts), 2):
    category = contexts[i+1].split("=")[1].strip()
    phrase_context_index[(phrase,contexts[i])] = category
#   print (phrase,contexts[i]), category

for line in sys.stdin:
  line_segments = line.split('|||')
  source = ['<s>' for x in range(order)] + line_segments[0].split() + ['</s>' for x in range(order)]
  target = ['<s>' for x in range(order)] + line_segments[1].split() + ['</s>' for x in range(order)]
  phrases = [ [int(i) for i in x.split('-')] for x in line_segments[2].split()]

  print "|||",

  for s1,s2,t1,t2 in phrases:
    s1 += order
    s2 += order
    t1 += order
    t2 += order

    phrase = reduce(lambda x, y: x+y+" ", target[t1:t2], "").strip()
    left_context = reduce(lambda x, y: x+y+" ", target[t1-order:t1], "")
    right_context = reduce(lambda x, y: x+y+" ", target[t2:t2+order], "").strip()
    context = "%s<PHRASE> %s" % (left_context, right_context)

    label = phrase_context_index[(phrase,context)]
    print "%s-%s:X%s" % (t1-order,t2-order,label),
  print