[keras] fit vs fit_generator

keras에서 모델을 설계하고 학습(fitting)할 때 사용하는 메소드가 fit과 fit_generator가 있습니다.

 

두개가 다른 것은 아나 어떤 차이인지 정확히 몰라 구글에 검색을 해보니 이미 쉽게 정리해둔 글이 있었습니다.

 

간단하게 정리하자면 fit()은 scikit-learn의 fit메소드와 유사하다고 합니다. 저희가 학습에 사용할 데이터 x와 y 전체를 한번에 입력으로 사용한다고 합니다. 그만큼 memory를 많이 사용하겠지요.

 

fit_generator()는 파이썬의 generator를 사용한 것으로 대용량의 데이터를 효율적으로 학습하기 위한 것이라고 합니다. 

파이썬의 제네레이터를 통해 형성된 데이터들을 batch-by-batch로 학습하는 방법입니다.

이 메소드는 CPU를 parallel하게 사용할 때 효율적이라고 합니다.