AdaptiveMotorControlLab / CEBRA

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA
https://cebra.ai
Other
875 stars 66 forks source link

Pass showlegend/template kwargs to plotly #122

Closed stes closed 7 months ago

stes commented 7 months ago

Fix #121

nastya236 commented 7 months ago

I think there is still a problem because of how Plotly interprets the color parameter in the marker dictionary, especially when it's linked with discrete labels. I think to how discrete labels the only solution it iteratively go through the unique labels and and the trace to the scetter plot. I made a PR here (branched from @stes branch).

plot_embedding_interactive(embedding, labels, markersize=10, showlegend=True)
labels = array([5, 8, 5, 4, 6, 8, 2, 3, 1, 7, 2, 9, 7, 3, 5, 6, 5, 9, 8, 4, 4, 7,
       2, 0, 1, 9, 5, 7, 5, 4, 9, 5, 1, 5, 3, 6, 4, 6, 3, 6, 3, 6, 0, 0,
       4, 7, 9, 1, 1, 5, 5, 4, 0, 0, 0, 6, 8, 9, 9, 4, 4, 4, 5, 2, 4, 2,
       7, 7, 6, 3, 5, 8, 0, 3, 6, 7, 9, 9, 1, 6, 9, 7, 4, 8, 7, 6, 5, 4,
       4, 0, 2, 9, 6, 6, 9, 2, 3, 6, 7, 6])

Screenshot 2024-01-04 at 12 51 45

I proposed a solutionhere: PR#123

stes commented 7 months ago

spelling error, fixed