c6b386f65b6295d4fbe691f7eb78ec4982009ef9,jieba/posseg/viterbi.py,,get_top_states,#Any#Any#,5

Before Change

MIN_INF = float("-inf")

def get_top_states(t_state_v, K=4):
    topK = sorted(t_state_v.items(), key=operator.itemgetter(1), reverse=True)[:K]
    return [x[0] for x in topK]

def viterbi(obs, states, start_p, trans_p, emit_p):
    V = [{}] //tabular
    mem_path = [{}]

After Change

MIN_INF = float("-inf")

def get_top_states(t_state_v, K=4):
    return sorted(t_state_v, key=t_state_v.__getitem__, reverse=True)[:K]

def viterbi(obs, states, start_p, trans_p, emit_p):
    V = [{}] //tabular
    mem_path = [{}]

Instances