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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
import argparse
import re
import sys
from collections import Counter
import json
def para_to_chunks(text, char_level_pred):
chunks = []
preds = []
lastchunk = ''
lastpred = ''
for idx in range(len(text)):
if re.match('^\w$', text[idx], flags=re.UNICODE):
lastchunk += text[idx]
else:
if len(lastchunk) > 0 and not re.match('^\W+$', lastchunk, flags=re.UNICODE):
chunks += [lastchunk]
assert len(lastpred) > 0
preds += [int(lastpred)]
lastchunk = ''
if not re.match('^\s$', text[idx], flags=re.UNICODE):
# punctuation
chunks += [text[idx]]
assert len(lastpred) > 0
preds += [int(char_level_pred[idx])]
else:
# prepend leading white spaces to chunks so we can tell the difference between "2 , 2" and "2,2"
lastchunk += text[idx]
lastpred = char_level_pred[idx]
if len(lastchunk) > 0:
chunks += [lastchunk]
preds += [int(lastpred)]
return list(zip(chunks, preds))
def paras_to_chunks(text, char_level_pred):
return [para_to_chunks(re.sub('\s', ' ', pt.rstrip()), pc) for pt, pc in zip(text.split('\n\n'), char_level_pred.split('\n\n'))]
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('plaintext_file', type=str, help="Plaintext file containing the raw input")
parser.add_argument('--char_level_pred', type=str, default=None, help="Plaintext file containing character-level predictions")
parser.add_argument('-o', '--output', default=None, type=str, help="Output file name; output to the console if not specified (the default)")
args = parser.parse_args()
with open(args.plaintext_file, 'r') as f:
text = ''.join(f.readlines()).rstrip()
text = '\n\n'.join([x for x in text.split('\n\n')])
if args.char_level_pred is not None:
with open(args.char_level_pred, 'r') as f:
char_level_pred = ''.join(f.readlines())
else:
char_level_pred = '\n\n'.join(['0' * len(x) for x in text.split('\n\n')])
assert len(text) == len(char_level_pred), 'Text has {} characters but there are {} char-level labels!'.format(len(text), len(char_level_pred))
output = sys.stdout if args.output is None else open(args.output, 'w')
json.dump(paras_to_chunks(text, char_level_pred), output)
output.close()
|