导入后如何在TFGO中与TF模型进行交互

我遵循https://www.tensorflow.org/tutorials/keras/text_classification上的TensorFlow教程并保存了模型。

我能够使用tfgo库成功导入Go:

package main

import (
    "fmt"

    tg "github.com/galeone/tfgo"
    tf "github.com/tensorflow/tensorflow/tensorflow/go"
)

func main() {
    model := tg.Loadmodel("movie_reviews",[]string{"serve"},nil)
    root := tg.NewRoot()
    t := tg.NewTensor(root,tg.Const(root,[3]int32{1,2,3}))
    fake,_ := tf.NewTensor([3]int32{1,3})
    model.Exec([]tf.Output{t.Output},map[tf.Output]*tf.Tensor{
        model.Op("input",0): fake,})
    fmt.Println(model)
}

但是现在我不知道如何与之交互。在Python中,您拥有所有这些模型方法,即predictevaluate等。

使用Go绑定,您似乎需要知道确切的操作名称才能与之交互?

我该如何找到答案?

ch446909360 回答:导入后如何在TFGO中与TF模型进行交互

是的,使用Go绑定和tfgo,您必须知道确切的操作名称。

顺便说一下,使用saved_model_cli CLI工具(随TensorFlow Python软件包一起提供)很容易获得所有名称。

saved_model_cli show --all --dir <path of your SavedModel>

对于SavedModel中可用的每个标记,它将为您提供所需的所有信息。对于您的情况,您必须使用键“ serve”或“ serving_default”查看signature_def

免责声明:我是tfgo软件包的作者。在“使用TensorFlow 2.0的动手神经网络”一书的第10章中,我还介绍了该主题,该节专门介绍SavedModel序列化格式。

本文链接:https://www.f2er.com/3163012.html

大家都在问