모델 저장

학습한 모델을 저장하는 방법은 2가지가 있다. 파이썬 객체 저장 : Pickle & Joblib 라이브러리 사용 Keras 모델 저장/로드 함수 사용 Pickle과 Joblib의 사용법은 거의 유사한데 Joblib의 사용법이 더 쉽기 때문에 Joblib을 기준으로 설명하겠다. [ Joblib 라이브러리 사용 ] 저장 : dump(객체, 저장경로) 확장자는 .pkl로 해준다. joblib.dump(data, 'data.pkl') 머신러닝 모델 저장 예시 불러오기 : load(불러올객체) loaded_data = joblib.load('data.pkl') 머신러닝 모델 불러오기 예시 [ Keras 모델 저장/로드 함수 사용 ] 케라스로 만든 모델의 경우에는 케라스에서 제공하는 함수를 사용해서 모델을 저장/로..
ImageDataGenerator Keras에서 이미지 파일을 쉽게 전처리하여 학습시킬 수 있도록 제공해주는 클래스이다. 뿐만 아니라 이미지 데이터에 여러 변형을 주어 원래 있는 학습 데이터셋보다 더 많은 양의 데이터를 학습시키는 것과 같은 효과를 낼 수 있는 이미지 증강(Image Argumentation)도 가능하게 해준다. 불러오기 import tensorflow as tf from tensorflow import keras from keras.preprocessing.image import ImageDataGenerator ImageDataGenerator() 인스턴스 생성 train_datagen = ImageDataGenerator( rescale=1. / 255, # 이미지의 픽셀 값을 조정..
kybeen
'모델 저장' 태그의 글 목록