keras中可以使用
from keras.utils import plot_model
plot_model(model,to_file='model_auth.png',show_shapes=True)
#show_shapes=True可以把輸入輸出的shape一起打印
注意,最好是給每個層命名,命名好之后打印出來的才會帶名字。程序運行的時候也有一定的指示作用。
keras中可以使用
from keras.utils import plot_model
plot_model(model,to_file='model_auth.png',show_shapes=True)
#show_shapes=True可以把輸入輸出的shape一起打印
注意,最好是給每個層命名,命名好之后打印出來的才會帶名字。程序運行的時候也有一定的指示作用。
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。