gobert: github.com/buckhx/gobert/model/estimator Index | Files

package estimator

import "github.com/buckhx/gobert/model/estimator"

Package estimator is a utility method for interacting with tf models. *** Experimental *** This package is meant ot be a pseudo-port of the python Estimator API

Package estimator is a utility method for interactinfg with tf models This package is meant ot be a pseudo-port of the python Estimator API

Index

Package Files

estimator.go predict.go

type Estimator Uses

type Estimator interface {
    /*
    	Trainer
    	Evaluator
    	Exporter
    */
    Predictor
}

Estimator matches the tf, p

type InputFunc Uses

type InputFunc func(m *tf.SavedModel) map[tf.Output]*tf.Tensor

InputFunc matches feeds in sessions

type ModelFunc Uses

type ModelFunc func(m *tf.SavedModel) ([]tf.Output, []*tf.Operation)

ModelFunc the returned params match fetches & targets from the API

type Predictor Uses

type Predictor interface {
    Predict(InputFunc) ([]*tf.Tensor, error)
}

Predictor creates tensors for prediction

func NewPredictor Uses

func NewPredictor(m *tf.SavedModel, fn ModelFunc) Predictor

NewPredictor creates a new Predictor in lieu of a full estimator

Package estimator imports 1 packages (graph) and is imported by 1 packages. Updated 2019-07-30. Refresh now. Tools for package owners.