predict

package
v0.0.1 Latest Latest
Warning

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

Go to latest
Published: Nov 12, 2021 License: Apache-2.0 Imports: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Predictor

type Predictor struct {
	Model *tf.SavedModel
}

func NewPredictor

func NewPredictor(modeldir string) (Predictor, error)

declare method

func (Predictor) Predict

func (p Predictor) Predict(imagefile string, model *tf.SavedModel) (int, error)

Predict - Return index with highest probability

func (Predictor) PredictionsArr

func (p Predictor) PredictionsArr(imagefile string, model *tf.SavedModel) ([]float32, error)

PredictionsArr Return array of probabilities

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL