jittery long_term_emb

[dikang13]

Hi NerdsLab,

Thank you for making BAMS a public resource. I have tried playing with the mouse triplets dataset without tweaking any parameters/ hyperparameters. However, I obtain long term embeddings that are quite jittery (sometimes more so than the short term embeddings).

The overview.png in the package shows very smooth state traces, which is what I expected to see from plotting the long term embeddings. Would it be possible to publish an example notebook that you wrote to generate overview.png and other figures in your paper? Thank you for your help!

Candy

[dikang13]

Hi, I might have identified the issue.

Line 115 of mouse_triplets.py, which defines head_vy, seems to have a typo. Instead of head_vy = diff(head_center[..., 0]), maybe it should be head_vy = diff(head_center[..., 1])?