dattalab / jax-moseq

Other
5 stars 5 forks source link

Port in `lgssm_posterior_sample` function #6

Closed ezhang94 closed 1 year ago

ezhang94 commented 1 year ago

Closes #4

Scales for masked dynamics noise and masked observation noise are currently hardcoded in, i.e. in

I would suggest setting these as parameters that can be passed in, so that they can be appropriately set, relative to the data noise, but I think they should do for now.

Results from demo notebook, using example data

Fitted AR-HMM

image

Fitted keypoint-SLDS

image

calebweinreb commented 1 year ago

How did you choose 10 for the masked noise/dynamics? Why not something bigger like 1e6?