在Keras顺序模块中保存和加载Tensorflow概率层

from tensorflow import keras

model = keras.Sequential()
# 添加概率层到模型中
model.add(keras.layers.Dense(64, activation='relu', input_shape=(10,)))
model.add(keras.layers.Dense(64, activation='relu'))
model.add(keras.layers.Dense(1, activation='sigmoid'))

# 保存模型
model.save('model.h5')

from tensorflow import keras

# 加载模型
model = keras.models.load_model('model.h5')

# 使用加载的模型进行预测等操作