加载由TF v2中的估算器生成的SavedModel

如何解决加载由TF v2中的估算器生成的SavedModel

我使用TF2中的估算器BoostedTreesClassifier训练了一个模型，并将其保存为SavedModel。 我遇到一些问题，以了解如何在另一个脚本中加载此SavedModel并使用它来计算一些预测。 我发现了大量关于如何使用Keras或TF1来实现此目的的代码，但是没有关于TF2的实现。明确地说，我试图避免使用tf.compat.v1.xxx，因为我想在TF2中有更好的方法来实现这一点。

• 我期望我们可以从SavedModel创建一个估算器并直接查询它，但是找不到任何方法

• 我尝试着加载模型并获取图形：

    model = tf.saved_model.load(model_dir_path,["serve"])
    graph = model.graph

但这看上去很底层，我不确定如何执行图形（除了使用tf.compat.v1之外，我再次尝试避免这种情况）

如果您有任何想法，请告诉我，非常感谢！