3.6.5 fit( ) 메서드 이해하기
compile() 다음에는 fit() 메서드를 호출합니다. fit() 메서드는 훈련 루프를 구현합니다. 다음은 fit() 메서드의 주요 매개변수입니다.
• 훈련할 데이터(입력과 타깃): 일반적으로 넘파이 배열이나 텐서플로 Dataset 객체로 전달합니다. 다음 장에서 Dataset API에 대해 자세히 배우겠습니다.
• 훈련할 에포크(epoch) 횟수: 전달한 데이터에서 훈련 루프를 몇 번이나 반복할지 알려 줍니다.
• 미니 배치 경사 하강법의 각 에포크에서 사용할 배치 크기: 가중치 업데이트 단계에서 그레이디언트를 계산하는 데 사용될 훈련 샘플 개수를 말합니다.
코드 3-23 넘파이 데이터로 fit( ) 메서드 호출하기
= model.fit(
inputs, ➊
targets, ➋
=5, ➌
=128 ➍
)
➊ 입력 샘플(넘파이 배열)
➋ 훈련 타깃(넘파이 배열)
➌ 이 데이터에서 훈련 루프를 다섯 번 반복합니다.
➍ 훈련 루프는 128개의 샘플 배치로 이 데이터를 순회합니다.