CNN 알고리즘을 적용시켜 모델을 학습시키고 .save() 메서드를 사용하여 모델을 저장하는 과정 오류가 생겼다.
이 부분에서 오류가 생겼다.
오류 내용
UserWarning: You are saving your model as an HDF5 file via `model.save()`. This file format is considered legacy. We recommend using instead the native Keras format, e.g. `model.save('my_model.keras')`.
saving_api.save_model(
이 경고 메세지는 Keras에서 모델을 저장할 때 HDF5 형식을 사용하는 것에 대한 경고이다.
이 경고의 주요 내용은
- model.save() 메서드를 사용하여 모델을 HDF5 형식으로 저장하려고 합니다.
- HDF5 형식은 이제 Keras에서 고려되는 레거시(legacy) 파일 형식입니다.
- 대신 Keras의 네이티브 형식으로 모델을 저장하는 것이 권장됩니다.
오류가 뜨긴 하지만 문제없이 모델은 h5 형식으로 저장되는 걸 확인할 수 있다.
해결 방법
1. 파일 확장자를 .h5 또는 .hdf5 대신 .keras로 사용하여 모델을 저장한다.
model.save('my_model.keras')
2. 위의 방법으로 저장된 모듈은 나중에 load_model 함수를 사용하여 로드할 수 있다.
from tensorflow.keras.models import load_model
loaded_model = load_model('my_model.keras')
위 방식대로 했더니 학습 모드를 실행시켜도 오류가 나오지 않았다.
또한 모델이 아래와 같이 잘 저장되었다.
'개발공부 > Python' 카테고리의 다른 글
[Error] Python mediapipe 사용 중 오류: FileNotFoundError: The path does not exist 해결방법 (0) | 2023.09.09 |
---|---|
[Pytube] python 모듈 Pytube로 Youtube 비디오 다운로드하기 (1) | 2023.09.08 |
[Python] open cv 설치 오류 ERROR: Could not find a version that satisfies the requirement cv2 (from versions: none) (0) | 2023.08.23 |
[Python] Pandas를 이용한 레고 데이터셋 분석 (0) | 2023.08.20 |
[Python] matplotlib로 시간 흐름에 따른 개발 언어 인기도 분석하기 (0) | 2023.08.20 |