Tensorflow 이것저것 1

많은 데이터를 수집하고 전처리하고 모델에 넣고 결과가 나오기까지 많은 시행착오가 필요할 겁니다.
그렇기에 우리는 load_model을 통해 남들이 만들어 놓은 모델을 불러오기도 하는데
이때 저는 RBFNN을 불러와서 사용해보겠습니다.
(모델 출처 : https://github.com/raaaouf/RBF_neural_network_python)
그 때 겪은 시행착오 중 하나를 소개드리겠습니다.

from tensorflow.keras.models import load_model
RBFNN_model= load_model("RBFNN_model.h5", custom_objects={'RBFLayer': RBFLayer})

모델을 load하기 위해서 사용.
save된 모델을 불러와서 사용하려면 위 코드를 통해서 사용할 수 있습니다.

custom_object를 하지 않고 load_model을 하게 되면 아래와 같은 error가 발생하게 된답니다!

꼭! custom_objects={'RBFLayer': RBFLayer}를 붙이도록 하자!!!