GoogleColabでネットワーク図を出力する

Last updated : 2019-09-20

今日現在はこれでうごいた

# version check
import tensorflow as tf; print(tf.__version__)

# modeling
import tensorflow as tf
from keras.models import Sequential
from keras.layers.core import Dense, Activation

model = Sequential([
        Dense(512, input_shape=(784,)),
        Activation('sigmoid'),
        Dense(10),
        Activation('softmax')
        ])

# figure
from keras.utils import plot_model
# from keras.utils.vis_utils import plot_model

plot_model(model, to_file='model.png')