visualkeras: Keras DNN Architecture Visualization PyPI package

Follow the full discussion on Reddit.
Hello reddit! I have published my first pypi package called visualkeras. So far it supports plotting visualizations in a stacked style, like commonly used for CNNs. I added many parameters so you can tweak it as you need and I will try to add more views (e.g. a graph mode for dense nets) so you can finally generate your architecture directly from keras / python.

Comments

There's unfortunately not much to read here yet...

Discover the Best of Machine Learning.

Ever having issues keeping up with everything that's going on in Machine Learning? That's where we help. We're sending out a weekly digest, highlighting the Best of Machine Learning.

Join over 900 Machine Learning Engineers receiving our weekly digest.

Best of Machine LearningBest of Machine Learning

Discover the best guides, books, papers and news in Machine Learning, once per week.

Twitter