加载由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之外,我再次尝试避免这种情况)

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

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)