| sent2id = dict() | |
| with open('sample300_en.txt', 'r') as f: | |
| i = 1 | |
| for line in f.readlines(): | |
| sent2id[line.replace('\n', '')] = i | |
| i += 1 | |
| with open('output_extractions.txt', 'r') as fin, open('compactIE_1.2_new.txt', 'w') as fout: | |
| for line in fin.readlines(): | |
| sentence, extraction, score = line.split('\t') | |
| sentId = sent2id[sentence] | |
| try: | |
| arg1 = extraction[extraction.index('<arg1>') + 6:extraction.index('</arg1>')] | |
| arg1 = arg1.strip() | |
| except: | |
| print("subject error!", extraction) | |
| arg1 = "" | |
| try: | |
| rel = extraction[extraction.index('<rel>') + 5:extraction.index('</rel>')] | |
| rel = rel.strip() | |
| except: | |
| print("predicate error!", extraction) | |
| rel = "" | |
| try: | |
| arg2 = extraction[extraction.index('<arg2>') + 6:extraction.index('</arg2>')] | |
| arg2 = arg2.strip() | |
| if arg2 == "": | |
| continue | |
| except: | |
| print("object error!", extraction) | |
| arg2 = "" | |
| print("{}\t{}\t{}\t{}".format(sentId, arg1, rel, arg2), file=fout) |