如何保存由要素图层组成的Keras模型？

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

# 定义输入层
inputs = Input(shape=(input_shape,))

# 添加要素图层
# ...

# 添加输出层
outputs = Dense(num_classes, activation='softmax')(x)

# 创建模型
model = Model(inputs=inputs, outputs=outputs)

model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])

model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, validation_data=(x_val, y_val))

model.save('model.h5')

from keras.models import load_model

model = load_model('model.h5')