Keras 모델

Keras로 배우는 Advanced Deep Learning

Zach Deane-Mayer

Data Scientist

Keras 모델

from tensorflow.keras.layers import Input, Dense
input_tensor = Input(shape=(1,))
output_tensor = Dense(1)(input_tensor)

Keras 모델

from tensorflow.keras.models import Model
model = Model(input_tensor, output_tensor)

모델 컴파일

model.compile(optimizer='adam', loss='mae')

모델 요약

model.summary()

Model: "model"
_________________________________________________________________
 Layer (type)                Output Shape              Param #   
=================================================================
 input_1 (InputLayer)        [(None, 1)]               0         

 dense (Dense)               (None, 1)                 2         

=================================================================
Total params: 2
Trainable params: 2
Non-trainable params: 0
_________________________________________________________________

Keras로 모델 시각화

input_tensor = Input(shape=(1,))
output_layer = Dense(1, name='Predicted-Score-Diff')
output_tensor = output_layer(input_tensor)
model = Model(input_tensor, output_tensor)
plot_model(model, to_file ='model.png')

from matplotlib import pyplot as plt
img = plt.imread('model.png')
plt.imshow(img)
plt.show()

연습해 봅시다!

Keras로 배우는 Advanced Deep Learning