You must install pydot (pip install pydot
) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) for plot_model to work.
Or for windows - winget install graphviz
Calculation Data - Train loss: 4.070760250091553 Validation loss: 4.5922698974609375 Test: loss 4.938761234283447