predictor

type Base struct {
	Framework dlframework.FrameworkManifest
	Model     dlframework.ModelManifest
	WorkDir   string
	Options   *options.Options
}

type Predictor interface {
	// Gets framework and model manifests
	Info() (dlframework.FrameworkManifest, dlframework.ModelManifest, error)
	// Gets predictor's Modality
	Modality() (dlframework.Modality, error)
	// Set desired output modality if not general
	SetDesiredOutput(modality dlframework.Modality)
	// Load model from manifest
	Load(ctx context.Context, model dlframework.ModelManifest, opts ...options.Option) (Predictor, error)
	// Returns the prediction options
	GetPredictionOptions() (*options.Options, error)
	// Returns the handle to features
	Predict(ctx context.Context, data interface{}, opts ...options.Option) error
	// Returns the raw inference result as go tensors
	ReadPredictedFeaturesAsMap(ctx context.Context) (map[string]interface{}, error)
	// Clears the internal state of a predictor
	Reset(ctx context.Context) error

	io.Closer
}