[BUG] Unable to use "graph_view" on a Sequential model (Keras version 2.0.0+)

paulgavrikov / visualkeras

Visualkeras is a Python package to help visualize Keras (either standalone or included in TensorFlow) neural network architectures. It allows easy styling to fit most needs. This module supports layered style architecture generation which is great for CNNs (Convolutional Neural Networks), and a graph style architecture, which works great for most models including plain feed-forward networks.

MIT License

564 stars 67 forks source link

Code to reproduce:

from keras import Sequential
from keras.layers import Dense, LSTM, Dropout, BatchNormalization

# Make sample sequential model. One input layer, one dense layer, one output layer
model = Sequential()
model.add(Dense(10, input_dim=3))
model.add(Dense(20))
model.add(Dropout(0.5))
model.add(BatchNormalization())
model.add(Dense(10))
model.add(Dropout(0.5))
model.add(BatchNormalization())
model.add(Dense(50))
model.add(Dropout(0.5))
model.add(BatchNormalization())
model.compile(optimizer='adam', loss='mse')

# Visualize the model
visualkeras.graph_view(model, to_file='output.png').show()

Tested Keras versions: 2.x

Will apply a fix soon. For now I'll keep this issue open and update the readme.md

paulgavrikov / visualkeras

[BUG] Unable to use "graph_view" on a Sequential model (Keras version 2.0.0+) #57