added k-means initialisation for Poisson HMMs

probml / dynamax

State Space Models library in JAX

https://probml.github.io/dynamax/

MIT License

634 stars 70 forks source link

Open anuj-sp opened 4 months ago

anuj-sp commented 4 months ago

Hi,

I have implemented the k-means initialization for the PoissonHMM emission rates.