【无标题】

1、自定义类模型ckpt

class AE(keras.Model):
    def __init__(self):
    ...
model = AE()

模型保存与加载

model.save_weights("modelsave/aednn.ckpt")
aednn=AE()
aednn.load_weights("modelsave/aednn.ckpt")

2、compile，Sequential h5

model = keras.Sequential([
    # layers.Dropout(0.5),
    layers.BatchNormalization(),
    layers.Dense(30, activation='relu'),
    # layers.Dense(20, activation='relu'),
    layers.Dense(5, activation='softmax')])

autoencoder = keras.Model(inputs=input_img, outputs=decoder_output)
autoencoder.compile(optimizer=tf.optimizers.Adam(lr),
                  loss='mse',
                  metrics=['accuracy'])

autoencoder.save('modelsave/autoencoder.h5')
autoencoder = tf.keras.models.load_model("modelsave/autoencoder.h5")

model.save('modelsave/dnn_cs.h5')