20241103_학습모델의 구성요소 및 저장방법

in 파이썬

개_고양이_구분_AI_만들기

1. 학습모델의 구성요소

  • 레이어
  • 손실함수 종류
  • 옵티마이저 종류
  • 훈련 후 가중치(w값)

2. 저장방법

  • 매우 간단한데, 학습시킨 모델 객체의 save메서드를 호출해주면 된다.
  • model.save(절대경로+ .keras) #(.keras는 텐서플로 기본 확장자)

3. 불러오는 방법

  • model.load(절대경로)

4. 체크포인트를 사용해서 epoch마다 가중치를 저장할 수도 있음

callbackMethod = tf.keras.callbacks.ModelCheckpoint(
    filepath = '절대경로{epoch}.weights.h5', 
    save_weights_only=True, # 해당 옵션 사용 시 확장자는 .weights.h5가 되어야 함
    save_freq='epoch'
)

...

model.fit(trainX,tranY,validation_data=(testX,testY), epochs=5,callbacks=[callbackMethod]) # 학습 시 콜백함수 추가


# 로드는 이렇게 ! 
# model.load_weights('체크포인트절대경로')