| skipped 640 lines |
641 | 641 | | data_array[data_array == 4624] = 3 |
642 | 642 | | data_array[data_array == 4625] = 4 |
643 | 643 | | # model = hmm.GaussianHMM(n_components=3, covariance_type="full", n_iter=10000) |
644 | | - | model = hmm.MultinomialHMM(n_components=3, n_iter=10000) |
| 644 | + | model = hmm.CategoricalHMM(n_components=3, n_iter=10000) |
645 | 645 | | # model.startprob_ = start_probability |
646 | 646 | | model.emissionprob_ = emission_probability |
647 | 647 | | model.fit(np.array([data_array], dtype="int").T, lengths) |
| skipped 1223 lines |