动态规划维特比 viterbi python实现

最新推荐文章于 2024-12-09 11:42:26 发布

原创最新推荐文章于 2024-12-09 11:42:26 发布 · 674 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法 #维特比 #viterbi

本文介绍了一种利用维特比算法进行拼音纠错的方法，通过动态规划找到最优路径，有效提高了拼音输入的准确性。文章提供了Python实现代码，并详细解释了算法的工作原理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

动态规划

最近在做关于拼音纠错的项目, 需要用到大量算分的步骤,笛卡尔积是在太慢, 在网上查一下,维特比比较好用, 于是找一找搜一搜原理
看到了这个仁兄的https://blog.youkuaiyun.com/athemeroy/article/details/79339546
于是动手写个python 实现

def viterbi(data, get_score, big=True):
    def most(some):
        if big:
            return max(some)
        else:
            return min(some)
    short_data = []
    for left_index, right_index in enumerate(range(1, len(data))):
        before = data[left_index]
        after = data[right_index]
        if left_index == 0:
            for before_one in before:
                for after_one in after:
                    k = before_one + '_' + after_one
                    short_data.append((get_score(k), after_one, [before_one, after_one]))
        else:
            temp_data = []
            for after_one in after:
                temp_short_data = []
                for before_one in short_data:
                    k = before_one[1] + '_' + after_one
                    temp_short_data.append((before_one[0] + get_score(k), after_one, before_one[2] + [after_one]))
                temp_data.append(most(temp_short_data))
            short_data = temp_data
    return max(short_data)[2]


if __name__ == '__main__':
    data = [
        ['a'],
        ['b1', 'b2', 'b3'],
        ['c1', 'c2', 'c3'],
        ['d1', 'd2', 'd3'],
        ['e'],
    ]


    def get_score(k):
        try:
            relation_score = {
                'a_b1': 6, 'a_b2': 7, 'a_b3': 5, 'b1_c1': 5, 'b1_c2': 6, 'b1_c3': 9,
                'b2_c1': 4, 'b2_c2': 3, 'b2_c3': 7, 'b3_c1': 4, 'b3_c2': 6, 'b3_c3': 6, 'c1_d1': 7, 'c1_d2': 8, 'c1_d3': 3,
                'c2_d1': 5, 'c2_d2': 4, 'c2_d3': 3, 'c3_d1': 5, 'c3_d2': 7, 'c3_d3': 6, 'd1_e': 4, 'd2_e': 8, 'd3_e': 5, }
            score = relation_score[k]
        except:
            score = 0
        return score

    print(viterbi(data=data, get_score=get_score, big=False))