predict

Documentation

Index

• Variables
• type Predictor
  • func NewPredictor(modeldir string) (Predictor, error)
  • func (p Predictor) Predict(imagefile string, model *tf.SavedModel) (int, error)
  • func (p Predictor) PredictWithDeviation(imagefile string, model *tf.SavedModel) (classpred int, deviation float32, err error)
  • func (p Predictor) PredictionsArr(imagefile string, model *tf.SavedModel) ([]float32, error)

Variables

var ImageHeight int32 = 128

var ImageWidth int32 = 128

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) PredictWithDeviation

func (p Predictor) PredictWithDeviation(imagefile string, model *tf.SavedModel) (classpred int, deviation float32, err error)

Predict - return class prediction and standard deviation

func (Predictor) PredictionsArr

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

PredictionsArr Return array of probabilities