keras 모델 저장

skyepodium·2021년 5월 19일
0

1. 코드

생성한 모델을 파일로 저장은 다음과 같이 합니다.

from tensorflow import keras
from keras.layers import SimpleRNN, Dense, Input
from keras.models import Model


# 모델 생성
Inputs = Input(shape=(28, 28))
x1 = SimpleRNN(64, activation="tanh")(Inputs)
x2 = Dense(10, activation="softmax")(x1)
model = Model(Inputs, x2)

# 모델 저장 경로
SAVE_PATH = "./model.h5"

# 모델 저장
model.save(SAVE_PATH)

# 모델 불러오기
new_model = keras.models.load_model(SAVE_PATH)

# 모델 요약 정보 확인
new_model.summary()

2. 여담

.h5 파일이 16진수로 적혀있길래 hex 에디터로 열어보니 json 도 있고... 그냥 그렇다구

profile
callmeskye

0개의 댓글