https://platoblockchain.com/plato-data/how-to-visualize-neural-network-architectures-in-python/