我遵循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中,您拥有所有这些模型方法,即predict
,evaluate
等。
使用Go绑定,您似乎需要知道确切的操作名称才能与之交互?
我该如何找到答案?