tfmodel

package

v0.12.13 Latest Latest Go to latest Published: Sep 18, 2023 License: Apache-2.0 Imports: 13 Imported by: 0

Details

This section is empty.

View Source

var TFSessionPanicDuration time.Duration = 1 * time.Minute

func Export(session *tf.Session, graph *tf.Graph, name string) (interface{}, error)

Export attempts to pull the embedded lookup table from the Tensorflow graph.

func MatchOperation(graph *tf.Graph, name string) string

MatchOperation will attempt to locate an LookupTableExportV2 operation that is associated to the provided name.

func RunExport(session *tf.Session, exportOp *tf.Operation) (interface{}, error)

RunExport runs export

func Signature(model *tf.SavedModel) (*domain.Signature, error)

Signature searches the Tensorflow operation graph for inputs and outputs.

type Evaluator struct {
	domain.Signature
	// contains filtered or unexported fields
}

Evaluator represents evaluator

func NewEvaluator(id string, signature *domain.Signature, session *tf.Session) *Evaluator

NewEvaluator creates new evaluator

func (e *Evaluator) Close() error

Close closes evaluator

func (e *Evaluator) Evaluate(params []interface{}) ([]interface{}, error)

Evaluate evaluates model