In my ongoing PyTorch journey, I know I will want to visualize neural networks that I study.
Which tool to use?
Manim, which is used by the famous YouTube Channel 3Blue1Brown, would be great for this.
BUT, I have a ton more experience doing such things with Visual Python, and it's intuitive to this "physics and data based" modeler.
So, I put together this initial Python class for a starting point.
I am eager to add dynamic indicators for weight changes during training and indicators on the neurons of activation values and such.
If you want to follow the repo for this, it's here → Neural_Net_Visualizer.
Clone the repo, and I hope to be making updates to it ASAP.
I've attempted to make a simple input scheme to keep this tool extremely flexible.
I hope it helps you to "ABV" in your neural network studies.
Until next time,
Thom
Comments
Top
New
No posts