[P] Neural Network Model Builder & Visualiser | Netbrix.ml
I’m sure there are existing tools similar to this in existence, so I wanted to keep it as simple as possible and not try to over-engineer it. It’s meant to be easy and simple to use!
It has some cool features at the moment, like the ability to parse
(Alternatively you can paste in Keras JSON definitions from for example
Once that simple model definition is imported it will be parsed by the app to create the following neat visualisation:
From here, you can make all the typical changes you would want to make to a model, including changing/adding layer parameters, adding new layers, changing the order of layers and removing layers, all without having to rely on Google to find the names of layers or their attributes. It can then be easily exported (or copied to the clipboard) with one click as either Python Keras code or a JSON spec which can be imported into Keras.
It also has some nice features for building models from scratch, like the ability to add blocks of layers that come up in models frequently. Often it’s easy to forget the exact optimal order of layers for say a Convolutional block e.g. should pooling come before dropout or vice versa, and what about BatchNorm? Having preconfigured blocks of layers to choose from when building a model helps with this.
On top of that, just having an easily indexable list of layers is useful in itself.
There is also a host of existing model architectures for a variety of machine learning tasks/datasets to explore and this is one of the most helpful features for me personally. Having an easy and centralised way to access a bunch of existing standard model architectures to take inspiration from is really useful! Rather than creating a model from scratch, you can find an existing model on the site, change say the input shape and a few of the hyperparameters and export it as working Keras code in just a few clicks. It’s also great for learning about the different architectures commonly used in building models.
So, I’d love to get some feedback on this project! Is it useful? Do you like the design? I’m open to criticism, this is primarily for me to learn 😁
Thanks for reading!