estimator

package
v0.0.0-...-265756f Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jul 31, 2019 License: MIT Imports: 1 Imported by: 2

Documentation

Overview

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

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Estimator 

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

Estimator matches the tf, p

type InputFunc 

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

InputFunc matches feeds in sessions

type ModelFunc 

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

ModelFunc the returned params match fetches & targets from the API

type Predictor 

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

Predictor creates tensors for prediction

func NewPredictor 

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

NewPredictor creates a new Predictor in lieu of a full estimator

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.