keras.utils.vis_utils
모듈은 Keras 모델을 (graphviz
를 사용하여)그리는 유틸리티 함수를 제공합니다.
다음과 같이 모델의 그래프를 그리고 파일에 저장할 수 있습니다:
from keras.utils import plot_model
plot_model(model, to_file='model.png')
plot_model
은 두 개의 선택적인 인수를 가집니다:
show_shapes
(기본값은 False) 그래프 상에 출력 형태를 보여줄지 여부를 조정합니다.show_layer_names
(기본값은 True) 그래프 상에 계층명을 보여줄지 여부를 조정합니다.pydot.Graph
객체를 직접 받아서 표현할 수도 있습니다. 예를 들어 다음과 같이 ipython notebook에 보이게 할 수 있습니다:
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
SVG(model_to_dot(model).create(prog='dot', format='svg'))
이 문서는 Keras의 Visualization을 번역한 것입니다.