딥러닝 : fit 함수에서 배치 사이즈 (batch_size= )와 에포크(epochs=)

우선 필요한 라이브러리들을 import 해줍니다.

import tensorflow as tf
from tensorflow import keras
from keras.models import Sequential
from keras.layers import Dense

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

컴파일이 끝난 후에 학습을 할 때 위와 같은 코드로 학습을 실시 한다.

 

epoch

• 한 번의 epoch는 신경망에서 전체 데이터 셋에 대해 forward pass/backward pass 과정을 거친 것을 말함. 즉, 전체 데이터 셋에 대해 한 번 학습을 완료한 상태
• 전체 데이터를 몇 번 반복할 것인가에 대한 수를 epochs= 뒤에 적어준다.

 

batch_size(배치사이즈)

Gradient Descent 를 한번 계산하기 위한 학습 데이터의 개수 

 

실행 결과