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

[Deep Learning] epoch, batch_size 설정

by coding_su 2022. 12. 27.

📝딥러닝 에포크와 배치사이즈

batch_size= n : 한번 학습할때 몇개의 데이터를 넣어 학습할지 값을 입력

메모리의 한계와 속도 저하 때문에 대부분의 경우에는 한 번의 epoch에서 모든 데이터를 한번에 집어넣을 수 없다

 그래서 데이터를 나누어 주는데 이때 몇 번 나누어 주는가를 iteration, 각 iteration마다 주는 데이터 사이즈를 batch size라고함

 

epochs= n : 해당 데이터를 몇번 반복해서 학습할지 값을 입력

※ 한 번의 epoch는 신경망에서 전체 데이터 셋에 대해 forward pass/backward pass 과정을 거친 것을 말함

 즉, 전체 데이터 셋에 대해 한 번 학습을 완료한 상태

model.fit(X_train, y_train, batch_size= 10, epochs= 20)

댓글