369 B
369 B
API: HMM
from optimizr import HMM
model = HMM(n_states=2)
model.fit(X, n_iterations=100, tolerance=1e-6)
states = model.predict(X)
logp = model.score(X)
Parameters
n_states: number of hidden regimesfit(X, n_iterations=100, tolerance=1e-6): train with Baum-Welchpredict(X): Viterbi decoding →np.ndarrayscore(X): log-likelihood