保存和加载keras子类模型

import tensorflow as tf
from tensorflow import keras

class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel, self).__init__()
        self.dense = tf.keras.layers.Dense(64, activation='relu')
        self.dropout = tf.keras.layers.Dropout(0.5)
        self.output_layer = tf.keras.layers.Dense(10, activation='softmax')

    def call(self, inputs):
        x = self.dense(inputs)
        x = self.dropout(x)
        return self.output_layer(x)

model = MyModel()

model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(x_train, y_train, epochs=10, validation_data=(x_val, y_val))

tf.keras.models.save_model(model, 'path/to/save/model.h5')

loaded_model = tf.keras.models.load_model('path/to/save/model.h5', compile=False)

predictions = loaded_model.predict(x_test)