先导入包
import jieba
import numpy as np
from hmmlearn import hmm
states = ["box1", 'box2', 'box3'] #隐藏状态
observations = ["red", "white"] #观测状态
start_probability = np.array([0.2, 0.4, 0.4]) #模型参数
transition_probability = np.array([[0.5, 0.2, 0.3], [0.3, 0.5, 0.2],[0.2, 0.3, 0.5]])
emission_probability = np.array([[0.5, 0.5], [0.4, 0.6],[0.7, 0.3]])
n_states = len(states)
n_observations = len(observations)
model = hmm.MultinomialHMM(n_components = n_states)
model.startprob_ = start_probability
model.transmat_ = transition_probability
model.emissionprob_ = emission_probability
#1.概率计算问题
logprob = model.score(np.array([[0, 1, 0]]).T)
prob = np.exp(logprob)
print(prob)
>>0.13021800000000003
#2.解码问题
model.decode(np.array([[0, 1, 0]]). T, algorithm = "viterbi")
>>(-4.219907785197447, array([2, 2, 2]))
a = model.decode(np.array([[0, 1