mle-infrastructure / mle-toolbox

Lightweight Tool to Manage Distributed ML Experiments 🛠
https://mle-infrastructure.github.io/mle_toolbox/toolbox/
MIT License
3 stars 1 forks source link

Animation Visualizer Functions #41

Closed RobertTLange closed 2 years ago

RobertTLange commented 3 years ago

I want to have a set of standard animation helper functions to be added to utils/visualize. Refactor the subdirectory into static and dynamic visualizations. They should seamlessly integrate with the loading of the loggers. The particular 'animators' should include:

We need to establish input shape conventions for this. E.g. data should come in the following shape (time, x, y) for the heatmaps.

Note to Rob: Start by refactoring the code written for the swarm prototypes (cellular automata, lines). Make general.