keras.layers.Input layer not supported

[windisch]

The following code

from keras.layers import (
    Input,
    Dense,
)
from keras.models import Model

from ann_visualizer.visualize import ann_viz

layer_in = Input(shape=(10, ))
layer_out = Dense(2, activation="linear")(layer_in)

model = Model(inputs=layer_in, outputs=layer_out)

ann_viz(model)

will raise the following exception:

ValueError: ANN Visualizer: Layer not supported for visualizing

in ann_visualizer/visualize.py, line 123 under python3.6. Packages used

• ann-visualizer==2.5
• Keras==2.3.1