savedmodel

README

SavedModel Utils

Utils for making using of SavedModels in Go more similar to TensorFlow Serving

Documentation

Index

• Variables
• func GetSignatureDefFromReader(tags []string, signature string, r io.Reader) (*protobuf.SignatureDef, error)
• func NewPredictor(modelsDir string, name string, version int, signature string) (predict.Predictor, error)
• type Runner
  • func NewRunner(savedModel *tf.SavedModel, tags []string, signature string) (*Runner, error)
  • func NewRunnerWithSignature(savedModel *tf.SavedModel, signatureDef *protobuf.SignatureDef) (*Runner, error)
  • func (sr *Runner) Run(inputs map[string]*tf.Tensor, outputFilter []string) (map[string]*tf.Tensor, error)

Variables

var ErrSignatureNotFound = errors.New("no signature with given name found")

ErrSignatureNotFound error returned then given signature was not found from SavedModel

Functions

func GetSignatureDefFromReader

func GetSignatureDefFromReader(tags []string, signature string, r io.Reader) (*protobuf.SignatureDef, error)

GetSignatureDefFromReader returns SignatureDef from a given io.Reader

func NewPredictor

func NewPredictor(modelsDir string, name string, version int, signature string) (predict.Predictor, error)

NewPredictor returns a new predictor (predict.Predictor) for a given saved model folder path name and version

Types

type Runner

type Runner struct {
	// contains filtered or unexported fields
}

Runner transform Session.Run input and output based on given SignatureDef

func NewRunner

func NewRunner(savedModel *tf.SavedModel, tags []string, signature string) (*Runner, error)

NewRunner returns a new Runner for a given tag set, signature and tf.SavedModel

func NewRunnerWithSignature

func NewRunnerWithSignature(savedModel *tf.SavedModel, signatureDef *protobuf.SignatureDef) (*Runner, error)

NewRunnerWithSignature returns a new Runner for a given SignatureDef and tf.SavedModel

func (*Runner) Run

func (sr *Runner) Run(inputs map[string]*tf.Tensor, outputFilter []string) (map[string]*tf.Tensor, error)

Run executes contained SaveModel and uses given SignatureDef for mapping feeds and outputs