2

I was just testing keras neural networks, and I wanted to visualize my neural network. I followed a code example from a website, however the code raises and AssertionError -9. Here's the code.

from keras.models import Sequential
from keras.layers import Dense
from keras.utils.vis_utils import plot_model

model = Sequential()

model.add(Dense(4, input_shape=(2,), activation="sigmoid"))
model.add(Dense(3, input_shape=(4,), activation="sigmoid"))
model.add(Dense(1))

plot_model(model, to_file='model_plot.png', show_shapes=True, show_layer_names=True)

This is the error message. The error message

Really appreciate the help, thanks.

Splitwire Z
  • 63
  • 1
  • 8

0 Answers0