더북(TheBook)

생성된 네트워크를 활용하여 모형을 훈련시킵니다.

코드 7-7 모델 훈련

import time
units = 64
epochs = 4
t0 = time.time() ------ 모형의 실행 시간을 위해 시간을 t0에 저장

model = RNN_Build(units)
model.compile(optimizer=tf.keras.optimizers.Adam(0.001),
              loss=tf.losses.BinaryCrossentropy(),
              metrics=['accuracy'],
              experimental_run_tf_function=False) ------ ①

model.fit(train_data, epochs=epochs, validation_data=test_data, validation_freq=2) ------ ②

model.compile()에서는 다양한 하이퍼파라미터를 정의합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.