search

caten2 / Tripods2021UA

MIT License
3 stars 5 forks source link

Restore visualization for neural nets #23

Open caten2 opened 10 months ago

caten2 commented 10 months ago

In an early version of this project, Andrey Yao had added visualization for the NeuralNet objects with Graphviz. It would be nice to be able to generate images of neural nets with the following information:

  1. The architecture of the neural net.
  2. The activation function at each neuron. (Perhaps using a more informative __str__ method for the various subclasses of Operation.)
  3. Optionally, a depiction of what feeding forward looks like for some inputs into the neural net. This could be combined with the display method(s) for binary relations addressed in #22. This could show the value of each activation function on its inputs, providing there is enough space.