딥 러닝 모델 or 가중치(weight) 불러오는법

김우빈·2022년 6월 1일
0

JUST DEEP_HINT

목록 보기
2/6
post-thumbnail

모델 불러오기 코드

#기존에 학습한 전체 model 불러오기
from tensorflow.keras.models import load_model
model =load_model("./YOUR_MODEL_NAME.h5")

#weight가 포함된 model 불러오기 후, test accuracy 확인
metrics=model.evaluate(x_test, y_test, verbose=2)
print("Test accuracy:", metrics[1])

가중치(weight) 포함 전체 모델입니다.

가중치(weight) 불러오기 코드

#가중치 불러오기
model.load_weights('./YOUR_MODELS_WEIGHT_NAME.h5')
print("save succeed")

오늘도 깊은 하루 되세요^^

profile
DeepLearning, MLOps

0개의 댓글