维特比算法简单求解例子

这是统计学习方法中的一道题目，下面是维特比算法的代码实现：

import numpy as np

A = np.array([(0.5, 0.2, 0.3), (0.3, 0.5, 0.2), (0.2, 0.3, 0.5)])
B = np.array([(0.5, 0.5), (0.4, 0.6), (0.7, 0.3)])
P = np.array([(0.2, 0.4, 0.4)])
o = np.array([(0, 1, 0)])

T = 3
n = np.zeros((T, T))
m = np.zeros((T, T))
ans_t = np.zeros(T)

for i in range(3):   #i代表第几步
    for t in range(T):  #t代表状态
        if i == 0:  #初始状态的处理
            n[i][t] = P[i][t] * B[t][o[0][i]]
        else:
            for pre_t in range(T):
                n[i][t] = max(n[i][t], n[i-1][pre_t] * A[pre_t][t])
            n[i][t] = n[i][t] * B[t][o[0][i]]
            m[i][t] = np.array([n[i-1][j] * A[j, t] for j in range(T)]).argmax()

ans_t[T-1] = n[T-1, :].argmax() + 1
for t in range(T-2, -1, -1):
     ans_t[t] = m[t+1][int(ans_t[t+1]) - 1] + 1

print(ans_t)