你如何通过brat中的偏移来订购注释?

时间:2016-04-08 17:55:31

标签: annotations brat

使用快速注释器工具brat时,创建的注释文件似乎将按照用户执行注释的顺序显示注释。如果从文档的开头开始并最后执行注释,则注释自然会采用正确的偏移顺序。但是,如果您需要更早地在文档中添加并添加另一个注释,则输出.ann文件中注释的偏移顺序将会出现故障。

如何重新排列.ann文件,以便在完成后注释按偏移顺序排列?在brat中是否有一些允许你这样做的选项,或者是否必须编写自己的脚本才能执行?

1 个答案:

答案 0 :(得分:1)

什么都没听到,我写了一个python脚本来完成我的目标。首先,我按开始索引重新排序所有注释。其次,我重新排序标签号,使它们再次按升序排列。

import optparse, sys

splitchar1 = '\t'
splitchar2 = ' '

# for brat, overlapped is not permitted (or at least a warning is generated)
# we could use this simplification in sorting by simply sorting on begin.  it is
# probably a good idea anyway.
class AnnotationRecord:
    label = 'T0'
    type = ''
    begin = -1
    end = -1
    text = ''

    def __repr__(self):
        return self.label + splitchar1
             + self.type + splitchar2
             + str(self.begin) + splitchar2
             + str(self.end) + splitchar1 + self.text

def create_record(parts):
    record = AnnotationRecord()
    record.label = parts[0]
    middle_parts = parts[1].split(splitchar2)
    record.type = middle_parts[0]
    record.begin = middle_parts[1]
    record.end = middle_parts[2]
    record.text = parts[2]
    return record

def main(filename, out_filename):
    fo = open(filename, 'r')
    lines = fo.readlines()
    fo.close()

    annotation_records = []

    for line in lines:
        parts = line.split(splitchar1)
        annotation_records.append(create_record(parts))

    # sort based upon begin    
    sorted_records = sorted(annotation_records, key=lambda a: int(a.begin))

    # now relabel based upon the sorted order
    label_value = 1
    for sorted_record in sorted_records:
        sorted_record.label = 'T' + str(label_value)
        label_value += 1

    # now write the resulting file to disk
    fo = open(out_filename, 'w')
    for sorted_record in sorted_records:
        fo.write(sorted_record.__repr__())        
    fo.close()


#format of .ann file is T# Type Start End Text
#args are input file, output file
if __name__ == '__main__':
    parser = optparse.OptionParser(formatter=optparse.TitledHelpFormatter(), 
                                   usage=globals()['__doc__'],
                                   version='$Id$')
    parser.add_option ('-v', '--verbose', action='store_true',
                       default=False, help='verbose output')
    (options, args) = parser.parse_args()
    if len(args) < 2:
        parser.error ('missing argument')
    main(args[0], args[1])
    sys.exit(0)