2015-05-04 23:00:27 +00:00
|
|
|
"""Convert OntoNotes into a json format.
|
|
|
|
|
|
|
|
doc: {
|
|
|
|
id: string,
|
|
|
|
paragraphs: [{
|
|
|
|
raw: string,
|
|
|
|
sents: [int],
|
|
|
|
tokens: [{
|
|
|
|
start: int,
|
|
|
|
tag: string,
|
|
|
|
head: int,
|
|
|
|
dep: string}],
|
2015-05-26 17:28:29 +00:00
|
|
|
ner: [{
|
|
|
|
start: int,
|
|
|
|
end: int,
|
|
|
|
label: string}],
|
2015-05-04 23:00:27 +00:00
|
|
|
brackets: [{
|
|
|
|
start: int,
|
|
|
|
end: int,
|
2015-05-26 17:28:29 +00:00
|
|
|
label: string}]}]}
|
2015-05-06 14:31:00 +00:00
|
|
|
|
|
|
|
Consumes output of spacy/munge/align_raw.py
|
2015-05-04 23:00:27 +00:00
|
|
|
"""
|
2015-05-27 15:04:29 +00:00
|
|
|
from __future__ import unicode_literals
|
2015-05-04 23:00:27 +00:00
|
|
|
import plac
|
|
|
|
import json
|
|
|
|
from os import path
|
2015-05-27 15:04:29 +00:00
|
|
|
import os
|
2015-05-04 23:00:27 +00:00
|
|
|
import re
|
2015-05-27 15:04:29 +00:00
|
|
|
import codecs
|
2015-05-04 23:00:27 +00:00
|
|
|
|
|
|
|
from spacy.munge import read_ptb
|
|
|
|
from spacy.munge import read_conll
|
2015-05-26 17:28:29 +00:00
|
|
|
from spacy.munge import read_ner
|
2015-05-04 23:00:27 +00:00
|
|
|
|
|
|
|
|
|
|
|
def _iter_raw_files(raw_loc):
|
|
|
|
files = json.load(open(raw_loc))
|
|
|
|
for f in files:
|
|
|
|
yield f
|
|
|
|
|
|
|
|
|
2015-05-26 17:28:29 +00:00
|
|
|
def format_doc(file_id, raw_paras, ptb_text, dep_text, ner_text):
|
|
|
|
ptb_sents = read_ptb.split(ptb_text)
|
|
|
|
dep_sents = read_conll.split(dep_text)
|
2015-05-27 15:04:29 +00:00
|
|
|
if len(ptb_sents) != len(dep_sents):
|
|
|
|
return None
|
|
|
|
if ner_text is not None:
|
|
|
|
ner_sents = read_ner.split(ner_text)
|
|
|
|
else:
|
|
|
|
ner_sents = [None] * len(ptb_sents)
|
2015-05-04 23:00:27 +00:00
|
|
|
|
|
|
|
i = 0
|
2015-05-27 15:04:29 +00:00
|
|
|
doc = {'id': file_id}
|
|
|
|
if raw_paras is None:
|
|
|
|
doc['paragraphs'] = [format_para(None, ptb_sents, dep_sents, ner_sents)]
|
|
|
|
else:
|
|
|
|
doc['paragraphs'] = []
|
|
|
|
for raw_sents in raw_paras:
|
|
|
|
doc['paragraphs'].append(
|
|
|
|
format_para(
|
|
|
|
' '.join(raw_sents).replace('<SEP>', ''),
|
|
|
|
ptb_sents[i:i+len(raw_sents)],
|
|
|
|
dep_sents[i:i+len(raw_sents)],
|
|
|
|
ner_sents[i:i+len(raw_sents)]))
|
|
|
|
i += len(raw_sents)
|
2015-05-04 23:00:27 +00:00
|
|
|
return doc
|
|
|
|
|
|
|
|
|
2015-05-27 15:04:29 +00:00
|
|
|
def format_para(raw_text, ptb_sents, dep_sents, ner_sents):
|
|
|
|
para = {
|
|
|
|
'raw': raw_text,
|
|
|
|
'sents': [],
|
|
|
|
'tokens': [],
|
|
|
|
'brackets': []}
|
|
|
|
offset = 0
|
|
|
|
assert len(ptb_sents) == len(dep_sents) == len(ner_sents)
|
|
|
|
for ptb_text, dep_text, ner_text in zip(ptb_sents, dep_sents, ner_sents):
|
|
|
|
_, annot = read_conll.parse(dep_text, strip_bad_periods=True)
|
|
|
|
if ner_text is not None:
|
|
|
|
_, ner = read_ner.parse(ner_text, strip_bad_periods=True)
|
|
|
|
else:
|
|
|
|
ner = ['-' for _ in annot]
|
|
|
|
for token_id, (token, token_ent) in enumerate(zip(annot, ner)):
|
|
|
|
para['tokens'].append(format_token(offset, token_id, token, token_ent))
|
|
|
|
|
|
|
|
_, brackets = read_ptb.parse(ptb_text, strip_bad_periods=True)
|
|
|
|
for label, start, end in brackets:
|
|
|
|
if start != end:
|
|
|
|
para['brackets'].append({
|
|
|
|
'label': label,
|
|
|
|
'first': start + offset,
|
|
|
|
'last': (end-1) + offset})
|
|
|
|
offset += len(annot)
|
|
|
|
para['sents'].append(offset)
|
|
|
|
return para
|
|
|
|
|
|
|
|
|
|
|
|
def format_token(offset, token_id, token, ner):
|
|
|
|
head = (token['head'] + offset) if token['head'] != -1 else -1
|
|
|
|
return {
|
|
|
|
'id': offset + token_id,
|
|
|
|
'orth': token['word'],
|
|
|
|
'tag': token['tag'],
|
|
|
|
'head': head,
|
|
|
|
'dep': token['dep'],
|
|
|
|
'ner': ner}
|
|
|
|
|
|
|
|
|
|
|
|
def read_file(*pieces):
|
|
|
|
loc = path.join(*pieces)
|
|
|
|
if not path.exists(loc):
|
|
|
|
return None
|
|
|
|
else:
|
|
|
|
return codecs.open(loc, 'r', 'utf8').read().strip()
|
|
|
|
|
|
|
|
|
|
|
|
def get_file_names(section_dir, subsection):
|
|
|
|
filenames = []
|
|
|
|
for fn in os.listdir(path.join(section_dir, subsection)):
|
|
|
|
filenames.append(fn.rsplit('.', 1)[0])
|
|
|
|
return list(sorted(set(filenames)))
|
|
|
|
|
|
|
|
|
|
|
|
def main(onto_dir, raw_dir, out_loc):
|
|
|
|
# All but WSJ --- we do that separately, as we have the source docs
|
|
|
|
sections = [
|
|
|
|
'bc/cctv',
|
|
|
|
'bc/cnn',
|
|
|
|
'bc/msnbc',
|
|
|
|
'bc/p2.5_a2e',
|
|
|
|
'bc/p2.5_c2e',
|
|
|
|
'bc/phoenix',
|
|
|
|
'bn/abc',
|
|
|
|
'bn/cnn',
|
|
|
|
'bn/mnb',
|
|
|
|
'bn/nbc',
|
|
|
|
'bn/p2.5_a2e',
|
|
|
|
'bn/p2.5_c2e',
|
|
|
|
'bn/pri',
|
|
|
|
'bn/voa',
|
|
|
|
'mz/sinorama',
|
|
|
|
'nw/dev_09_c2e',
|
|
|
|
'nw/p2.5_a2e',
|
|
|
|
'nw/p2.5_c2e',
|
|
|
|
'nw/xinhua',
|
|
|
|
'pt/ot',
|
|
|
|
'tc/ch',
|
|
|
|
'wb/a2e',
|
|
|
|
'wb/c2e',
|
|
|
|
'wb/eng',
|
|
|
|
'wb/dev_09_c2e',
|
|
|
|
'wb/p2.5_a2e',
|
|
|
|
'wb/p2.5_c2e',
|
|
|
|
'wb/sel'
|
|
|
|
]
|
|
|
|
docs = []
|
|
|
|
for section in sections:
|
|
|
|
section_dir = path.join(onto_dir, 'data', 'english', 'annotations', section)
|
|
|
|
print section, len(docs)
|
|
|
|
for subsection in os.listdir(section_dir):
|
|
|
|
for fn in get_file_names(section_dir, subsection):
|
|
|
|
ptb = read_file(section_dir, subsection, '%s.parse' % fn)
|
|
|
|
dep = read_file(section_dir, subsection, '%s.parse.dep' % fn)
|
|
|
|
ner = read_file(section_dir, subsection, '%s.name' % fn)
|
|
|
|
if ptb is not None:
|
|
|
|
doc = format_doc(fn, None, ptb, dep, ner)
|
|
|
|
if doc is not None:
|
|
|
|
docs.append(doc)
|
|
|
|
# Now do WSJ, with source alignment
|
|
|
|
onto_dir = path.join(onto_dir, 'data', 'english', 'annotations', 'nw', 'wsj')
|
2015-05-04 23:00:27 +00:00
|
|
|
for i in range(25):
|
|
|
|
section = str(i) if i >= 10 else ('0' + str(i))
|
|
|
|
raw_loc = path.join(raw_dir, 'wsj%s.json' % section)
|
2015-05-23 15:39:04 +00:00
|
|
|
for j, (filename, raw_paras) in enumerate(_iter_raw_files(raw_loc)):
|
2015-05-04 23:00:27 +00:00
|
|
|
if section == '00':
|
|
|
|
j += 1
|
|
|
|
if section == '04' and filename == '55':
|
|
|
|
continue
|
2015-05-27 15:04:29 +00:00
|
|
|
ptb = read_file(onto_dir, section, '%s.parse' % filename)
|
|
|
|
dep = read_file(onto_dir, section, '%s.parse.dep' % filename)
|
|
|
|
ner = read_file(onto_dir, section, '%s.name' % filename)
|
|
|
|
if ptb is not None and dep is not None:
|
|
|
|
docs.append(format_doc(filename, raw_paras, ptb, dep, ner))
|
|
|
|
print 'nw/wsj', len(docs)
|
|
|
|
with open(out_loc, 'w') as file_:
|
|
|
|
json.dump(docs, file_, indent=4)
|
|
|
|
|
2015-05-04 23:00:27 +00:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
plac.call(main)
|