predict

Documentation

Index

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

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