def entity_linking_probs()

in entity_linking.py [0:0]

• 15 lines of code
• 6 McCabe index (conditional complexity)

def entity_linking_probs(v, distinct_slot_values, softmaxtemp, method_name, threshold=None):
    if method_name == 'bijaccard':
        distance = bi_jaccard_distance
    elif method_name == 'edit_distance':
        distance = leven_distance  # Case in-sensitive
    elif method_name == 'edit_distance_norm':
        distance = leven_distance_norm  # Case in-sensitive
    elif method_name == 'roberta_mnli':
        distance = roberta_mnli_distance
    elif method_name == 'average_three':
        distance = average_three_distance
    else:
        raise Exception(f'Invalid entity linking method name: {method_name}')

    probs = get_scores(v, distinct_slot_values, distance, threshold, softmaxtemp)
    return probs