Keras: Ошибка утверждения при построении модели - PullRequest
Новая
       56

Keras: Ошибка утверждения при построении модели

0 голосов
/ 30 апреля 2020

Я хочу построить свою модель Keras, используя функцию ниже. Каждый раз, когда я пытаюсь это сделать, мне выдается следующая ошибка AssertionError. Может кто-нибудь помочь? 

---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
<ipython-input-37-4e7b8789844d> in <module>
      9 
     10 model,_ = create_model()
---> 11 plot_keras_model(model, show_shapes=True, show_layer_names=False)

<ipython-input-37-4e7b8789844d> in plot_keras_model(model, show_shapes, show_layer_names)
      6 from keras.utils.vis_utils import model_to_dot
      7 def plot_keras_model(model, show_shapes=True, show_layer_names=True):
----> 8     return SVG(model_to_dot(model, show_shapes=show_shapes, show_layer_names=show_layer_names, expand_nested=True).create(prog='dot',format='svg'))
      9 
     10 model,_ = create_model()

/opt/conda/lib/python3.6/site-packages/keras/utils/vis_utils.py in model_to_dot(model, show_shapes, show_layer_names, rankdir, expand_nested, dpi, subgraph)
    180                             if not is_model(layer) and (
    181                                     not is_wrapped_model(layer)):
--> 182                                 assert dot.get_node(inbound_layer_id)
    183                                 assert dot.get_node(layer_id)
    184                                 dot.add_edge(pydot.Edge(inbound_layer_id,

AssertionError: 


import pydotplus as pydot
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
def plot_keras_model(model, show_shapes=True, show_layer_names=True):
    return SVG(model_to_dot(model, show_shapes=show_shapes, show_layer_names=show_layer_names).create(prog='dot',format='svg'))

model,_ = create_model()
plot_keras_model(model, show_shapes=True, show_layer_names=False)
...