How do I "flatten" a nested model?

[Isaac Gerg]

I have a model built with the functional API which is composed of other models.  This breaks some tools which analyze keras graphs.  Is there a function in keras which allows me to "flatten" these nested models so that when I do a model.layers, the list does not contain any keras.models?

Isaac

[fabia...@googlemail.com]

Hi Isaac,

I just had the same problem. Here is my solution for only one level of nesting. For multiple levels of nesting you could use recursion of course.

layers_flat = []

for layer in  model_nested.layers:

    try:

        layers_flat.extend(layer.layers)

    except AttributeError:

        layers_flat.append(layer)

model_flat = Sequential(layers_flat)

Fabian

[Isaac Gerg]

Thank you.