더북(TheBook)

이제 준비가 다 되었으니 다음처럼 에포크 5개를 학습시켜보자. 빠르게 실습하고자 에포크를 5개 지정한 것이고, 추후에는 에포크를 늘려 실습해보길 바란다.

>>> model.fit(x_train, y_train, epochs=5)
>>> model.evaluate(x_test, y_test, verbose=2)
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz
11493376/11490434 [==============================] - 2s 0us/step
Train on 60000 samples
Epoch 1/5
60000/60000 [==============================] - 5s 89us/sample - loss: 0.2934 - accuracy: 0.9146
Epoch 2/5
60000/60000 [==============================] - 4s 71us/sample - loss: 0.1421 - accuracy: 0.9581
Epoch 3/5
60000/60000 [==============================] - 4s 65us/sample - loss: 0.1075 - accuracy: 0.9669
Epoch 4/5
60000/60000 [==============================] - 5s 77us/sample - loss: 0.0877 - accuracy: 0.9732
Epoch 5/5
60000/60000 [==============================] - 4s 64us/sample - loss: 0.0761 - accuracy: 0.9762
10000/1 - 0s - loss: 0.0383 - accuracy: 0.9776
[0.07466363837001845, 0.9776]

에포크 5번 이후 최종 성능은 97.76% 정확도인 것으로 나타났다. 이제 인공 신경망의 구성을 다양하게 변경해보며 성능의 변화를 확인할 수 있다.

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