Skip to content

Saved model

jiangyi15@mails.ucas.ac.cn requested to merge saved_model into dev

Export Amplitude model to TensorFlow Saved Model format.

config = ConfigLoader("config,yml")
config.set_params("final_params.json")
config.save_tensorflow_model("model_dir")

Then the model can be used with TensorFlow only as

model = tf.saved_model.load("model_dir")
model(p1, p2, p3)

Merge request reports