search

mahyar-amiri / keras-visualizer

A Keras Model Visualizer
MIT License
25 stars 2 forks source link

Feature Request: change max neurons per layer #3

Closed swight-prc closed 1 year ago

swight-prc commented 3 years ago

It looks as though there's a maximum of 9 neurons displayed per Dense layer (at least - I haven't messed around too much with other layer types.) I would like to be able to update that per run, if that's possible.

Thanks!

mahyar-amiri commented 1 year ago

Hi, I apologize for the delay in replying.

The package has been updated. In the new version, you can simply change MAX_NEURONS for dense layer using code below.

my_custom_settings = {
    'MAX_NEURONS': 16
}

visualizer(model, file_format='png', settings=my_custom_settings)

you can also see other settings in README settings section.

Thanks for your feedback.