文本分析HMM（2）

先导入包

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