您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

TensorFlow:如何从SavedModel进行预测?

TensorFlow:如何从SavedModel进行预测?

加载图形后,它就可以在当前上下文中使用,您可以通过它馈入输入数据以获得预测。每个用例都有很大的不同,但是在代码添加内容如下所示:

with tf.Session(graph=tf.Graph()) as sess:
    tf.saved_model.loader.load(
        sess,
        [tf.saved_model.tag_constants.SERVING],
        "/job/export/Servo/1503723455"
    )

    prediction = sess.run(
        'prefix/predictions/Identity:0',
        Feed_dict={
            'Placeholder:0': [20.9],
            'Placeholder_1:0': [1.8],
            'Placeholder_2:0': [0.9]
        }
    )

    print(prediction)

在这里,您需要知道预测输入的名称。如果您没有给他们带来天真serving_fn,则它们认为Placeholder_n,这n是第n个功能

的第一个字符串参数sess.run是预测目标的名称。这将根据您的用例而有所不同。

其他 2022/1/1 18:37:06 有359人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶