ValueError：以 10 为底的 int() 的无效文字：'' 使用 ANN Visualizer

Question

我使用下面的 package 来可视化 lstm 架构。 但是，我收到以下错误：

代码

# Visualize the architecture

from ann_visualizer.visualize import ann_viz

ann_viz(lstm_model, view=True, title='LSTM Architecture')

Output

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-30-8cea540842d4> in <module>
      3 from ann_visualizer.visualize import ann_viz
      4 
----> 5 ann_viz(lstm_model, view=True, title='LSTM Architecture')

/usr/local/lib/python3.8/dist-packages/ann_visualizer/visualize.py in ann_viz(model, view, filename, title)
     40     for layer in model.layers:
     41         if(layer == model.layers[0]):
---> 42             input_layer = int(str(layer.input_shape).split(",")[1][1:-1]);
     43             hidden_layers_nr += 1;
     44             if (type(layer) == keras.layers.core.Dense):

ValueError: invalid literal for int() with base 10: ''

Answer 1

确保lstm_model是 Keras 顺序 model 视图（ pypi 上的 ann_visualizer）

似乎早在 2018 年就在github上发布了同样的问题