predictor

Documentation

Index

• func NewGeneralPredictor(model dlframework.ModelManifest, os ...options.Option) (common.Predictor, error)
• type GeneralPredictor
  • func (p *GeneralPredictor) Close() error
  • func (p *GeneralPredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)
  • func (p *GeneralPredictor) Modality() (dlframework.Modality, error)
  • func (p *GeneralPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error
  • func (p *GeneralPredictor) ReadPredictedFeaturesAsMap(ctx context.Context) (map[string]interface{}, error)
  • func (p *GeneralPredictor) Reset(ctx context.Context) error
  • func (p *GeneralPredictor) SetDesiredOutput(modality dlframework.Modality)

Functions

func NewGeneralPredictor

func NewGeneralPredictor(model dlframework.ModelManifest, os ...options.Option) (common.Predictor, error)

NewGeneralPredictor ...

Types

type GeneralPredictor

type GeneralPredictor struct {
	common.Base
	// contains filtered or unexported fields
}

GeneralPredictor ...

func (*GeneralPredictor) Close

func (p *GeneralPredictor) Close() error

Close ...

func (*GeneralPredictor) Load

func (p *GeneralPredictor) Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (common.Predictor, error)

Load ...

func (*GeneralPredictor) Modality

func (p *GeneralPredictor) Modality() (dlframework.Modality, error)

Modality ...

func (*GeneralPredictor) Predict

func (p *GeneralPredictor) Predict(ctx context.Context, data interface{}, opts ...options.Option) error

Predict ...

func (*GeneralPredictor) ReadPredictedFeaturesAsMap

func (p *GeneralPredictor) ReadPredictedFeaturesAsMap(ctx context.Context) (map[string]interface{}, error)

ReadPredictedFeaturesAsMap ...

func (*GeneralPredictor) Reset

func (p *GeneralPredictor) Reset(ctx context.Context) error

Reset ...

func (*GeneralPredictor) SetDesiredOutput

func (p *GeneralPredictor) SetDesiredOutput(modality dlframework.Modality)

This allows postprocess to use different output formats, however, the model has to output in a format that the desired modality postprocess can handle