tfmodel

Documentation

Index

• Variables
• func Export(session *tf.Session, graph *tf.Graph, name string) (interface{}, error)
• func MatchOperation(graph *tf.Graph, name string) string
• func RunExport(session *tf.Session, exportOp *tf.Operation) (interface{}, error)
• func Signature(model *tf.SavedModel) (*domain.Signature, error)
• type Evaluator
  • func NewEvaluator(id string, signature *domain.Signature, session *tf.Session) *Evaluator
  • func (e *Evaluator) Close() error
  • func (e *Evaluator) Evaluate(params []interface{}) ([]interface{}, error)

Variables

var TFSessionPanicDuration time.Duration = 1 * time.Minute

Functions

func Export

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

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

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

RunExport runs export

func Signature

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

Signature searches the Tensorflow operation graph for inputs and outputs.

Types

type Evaluator

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

Evaluator represents evaluator

func NewEvaluator

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

NewEvaluator creates new evaluator

func (*Evaluator) Close

func (e *Evaluator) Close() error

Close closes evaluator

func (*Evaluator) Evaluate

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

Evaluate evaluates model