본문 바로가기
인공지능/Deep Learning

[Deep Learning] epochs 횟수 조정

by coding_su 2022. 12. 29.

📝딥러닝 epochs 횟수 조정

epoch 횟수를 너무 많이 지정하면 학습한 데이터만 잘 맞추게 되는 오버피팅(Overfiting)이 된다
또 너무 적게 지정하면 학습이 덜 되어 언더피팅(Underfiting)이 된다

# 학습데이터 차트보기
plt.plot(epoch_history.history['loss'])
plt.plot(epoch_history.history['val_loss'])
plt.legend(['Train Loss', ' Validation Loss'])
plt.show()

# 밸리데이션데이터 차트보기
plt.plot(epoch_history.history['accuracy'])
plt.plot(epoch_history.history['val_accuracy'])
plt.legend(['Train Accuracy', ' Validation Accuracy'])
plt.show()

댓글