Function record_to_offset

evaluation.py:94–137

Find Role's offset using closest matched with trigger work. :param instance: :return:

(instance)

    return matched_list

def record_to_offset(instance):
    """
    Find Role's offset using closest matched with trigger work.
    :param instance:
    :return:
    """
    trigger_list = list()
    role_list = list()

    token_list = instance['text'].split()

    trigger_matched_set = set()
    for record in instance['pred_record']:
        event_type = record['type']
        trigger = record['trigger']
        matched_list = match_sublist(token_list, trigger.split())

        trigger_offset = None
        for matched in matched_list:
            if matched not in trigger_matched_set:
                trigger_list += [(event_type, matched)]
                trigger_offset = matched
                trigger_matched_set.add(matched)
                break

        # No trigger word, skip the record
        if trigger_offset is None:
            break

        for _, role_type, text_str in record['roles']:
            matched_list = match_sublist(token_list, text_str.split())
            if len(matched_list) == 1:
                role_list += [(event_type, role_type, matched_list[0])]
            elif len(matched_list) == 0:
                sys.stderr.write("[Cannot reconstruct]: %s %s\n" %
                                 (text_str, token_list))
            else:
                abs_distances = [abs(match[0] - trigger_offset[0])
                                 for match in matched_list]
                closest_index = np.argmin(abs_distances)
                role_list += [(event_type, role_type,
                               matched_list[closest_index])]

    return instance['text'], trigger_list, role_list

class Metric:
    def __init__(self):