predict

type Base struct {
	Framework dlframework.FrameworkManifest
	Model     dlframework.ModelManifest
}

type ImagePredictor struct {
	Base
	WorkDir string
}

type Predictor interface {
	// Gets framework information
	GetFramework() (dlframework.FrameworkManifest, error)
	// Load model from manifest
	Load(ctx context.Context, model dlframework.ModelManifest) (Predictor, error)
	// Downloads the features / symbol file / weights
	Download(ctx context.Context) error
	// Preprocess the data
	Preprocess(ctx context.Context, data interface{}) (interface{}, error)
	// Returns the features
	Predict(ctx context.Context, data interface{}) (*dlframework.PredictionFeatures, error)

	io.Closer
}