[Deep Learning] 딥러닝 모델 구조 시각화
keras를 활용한 딥러닝 모델 구조 시각화.
✅ 딥러닝 모델 구조 시각화
- keras에는 모델 구조를 이미지로 시각화할 수 있는 Tool이 존재
model.summary()
로 간단하게 확인할 수 있지만, 표로 더 직관적으로 확인할 수 있다.from tensorflow.keras.utils import plot_model
으로 불러와서plot_model(model)
으로 확인할 수 있다.plot_model
의 파라미터는 keras Document에서 확인하고 사용하면 된다.기본적으로show_shapes=True
하는게 보기는 좋은 듯 하다.
✅ 코드 예시
1
2
3
from tensorflow.keras.utils import plot_model
plot_model(model, show_shapes=True)
This post is licensed under CC BY 4.0 by the author.