keras에서 설명이있는 모델 저장

Question

설명이 포함 된 케 라스 모델 (간단한 문자열 일 수 있음)을 작성하여 파일에 포함 된 내용을 더 쉽게 추적 할 수 있습니다.

이 작업을 수행하는 간단한 방법이 있습니까? 내가 저장하고 내가하고 싶은 것이

model.save(filename) 
[...] 
model = keras.load_model(filename) 

와 모델을로드 할 수 있습니다,이 달성의 가장 간단한 방법 일 것입니다 무슨에

model.save(filename, description=string) 
[...] 
model, description = keras.load_model(filename) 

모든 아이디어의 효과에 뭔가?

Answer 1

현재 Keras에있는 model.save의 현재 버전에서는이를 수행 할 방법이 없습니다. loss function 및 metrics 및 load_model과 같은 모델의 구성을 저장하면 저장된 내용 만 반환됩니다. 이 문제에 대한

한 불쾌한 솔루션은 수동으로 Keras에서 클래스 models.py에 description 변수를 추가하고 결과적으로 save_model 및 load_model 기능에 코드의 필요한 줄을 추가한다. 다시 말하지만, 이것은 단지 끔찍한 해결책입니다!