Почему plot_model в Keras неправильно отображает модель? - PullRequest
0 голосов
/ 15 апреля 2020

Я хочу визуализировать свою нервную систему net. Поэтому я использую from tensorflow.keras.utils import plot_model и использую его так:

    model = Sequential()
    model.add(Dense(8, activation="relu"))
    model.add(Dense(1))
    plot_model(model, to_file="model.png", show_shapes=True)

Но когда я открываю график c, он выглядит так:

enter image description here

Что не так с моим кодом? Я не вижу никакой ошибки.

1 Ответ

1 голос
/ 15 апреля 2020

Причина в том, что модель не была построена, потому что она не знает свою входную форму. Либо укажите входную форму модели на первом слое, используя аргумент input_shape (или input_dim), либо альтернативно начните подгонку модели к некоторым данным, вызвав метод fit (чтобы входная форма могла быть автоматически выведена). Кроме того, как упомянуто @ xdurch0 в разделе комментариев, другой вариант - вызвать build метод модели и передать ему входную форму в качестве аргумента.

...