Discover this year's submissions to the Dataiku Frontrunner Awards and give kudos to your favorite use cases and success stories!READ MORE

Display model architecture on summary page for keras models

0 Kudos

I'd love to see the architecture used (as defined in the Architecture  section of VisualML) in the model Summary / Report page somehow.

When you're trying out different architectures you have to revert the design to a specific session each time you want to view the used architecture, which is a pain. And the added hazard of not realizing what session's settings the Design is set to when training a new model.

JUst the contents of build_model() and compile_model() functions would be enough.