Обучение с учителем HMM с гранатом

Question

Спасибо за вашу помощь, я новичок в этой теме.

У меня есть набор данных, состоящий из последовательности двоичных наблюдаемых переменных, связанных с меткой, как это.

trainset = [ [0,0,1,0],[0,0,1,0],[0,0,0,1],...,]
labels = [ 0,1,2,...,]
states = 3 # number of unique labels

Я пытаюсь обучить HMM таким образом

model = HiddenMarkovModel.from_samples(distribution = MultivariateGaussianDistribution, n_components = states,
                                       X= [trainset],
                                       labels = [labels],
                                       algorithm = 'labeled')

Когда я запускал эту модель, я получаю это исключение

Исключение игнорируется в: 'pomegranate.hmm.HiddenMarkovModel._labeled_summarize' KeyError: 3

Я понимаю, что исключение возникает, когда для состояния нет наблюдений -3-. Но, государство3есть в моих этикетках. Может кто-то мне помочь, пожалуйста? Спасибо