predictor

Documentation

Index

• Constants
• type Predictor
  • func NewPredictor(modelFile string) (*Predictor, error)
  • func (p *Predictor) DeletePredictor()
  • func (p *Predictor) GetPrediction() ([]float32, error)
  • func (p *Predictor) PredictProba(img image.Image) error

Constants

const (
	OutputSize = 1000
	BatchSize  = 1
)

Types

type Predictor

type Predictor struct {
	// contains filtered or unexported fields
}

func NewPredictor

func NewPredictor(modelFile string) (*Predictor, error)

func (*Predictor) DeletePredictor

func (p *Predictor) DeletePredictor()

func (*Predictor) GetPrediction

func (p *Predictor) GetPrediction() ([]float32, error)

func (*Predictor) PredictProba

func (p *Predictor) PredictProba(img image.Image) error