LSTM visualization [BUG]

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

586 stars 69 forks source link

LSTM visualization [BUG] #34

Closed MohamedNedal closed 1 year ago

MohamedNedal commented 1 year ago

Describe the bug I have trained a bidirectional LSTM model and I would like to visualize it using this tool. I got the errors in the attached screenshots.

Screenshots

Environment:

OS: Ubuntu 20.04.1 LTS (GNU/Linux 5.4.0-144-generic x86_64)
Jupyter Notebook on Python 3.6.13

paulgavrikov commented 1 year ago

Hi,this is a problem with the pypi hosted version, please install visualkeras directly from GitHub (see #8)