search

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

Fixed showlegend for plotly #123

Closed nastya236 closed 7 months ago

nastya236 commented 7 months ago

Fixed showlegend **kwargs in plotly scatter 3D. 

from cebra.integrations.plotly import plot_embedding_interactive
import numpy as np

embedding = np.random.randn(100, 3)
labels = np.random.randint(0, 10, 100)

plot_embedding_interactive(embedding, labels, markersize=10, showlegend=True)

Screenshot 2024-01-04 at 13 11 08

plot_embedding_interactive(embedding, labels, markersize=10)

Screenshot 2024-01-04 at 13 11 25

nastya236 commented 7 months ago

Of course, writing test for this. I made some modifications in the last commit. Update: tests are added and passed, ready for the review @stes