predictGeneral

Documentation

Index

• func CorePrediction(model *dl.ModelManifest, predictors []common.Predictor, settings Settings) (response string, err error)
• func InitCorePrediction() (err error)
• type Settings
  • func GetDefaultSettings() (o Settings)

Functions

func CorePrediction

func CorePrediction(model *dl.ModelManifest, predictors []common.Predictor, settings Settings) (response string, err error)

func InitCorePrediction

func InitCorePrediction() (err error)

Types

type Settings

type Settings struct {
	BatchSize                  int
	UseGPU                     bool
	GpuDeviceId                int
	GpuMetrics                 string
	PublishPredictions         bool
	PublishToDatabase          bool
	PartitionListSize          int
	SaveInferenceResult        bool
	FailOnFirstError           bool
	DisableFrameworkAutoTuning bool
	DesiredResultModality      string

	TracerServerAddr string
	TracePreprocess  bool
	TraceLevelName   string
	TraceLevel       tracer.Level
	TracerAddress    string

	DatabaseAddress      string
	DatabaseName         string
	DatabaseEndpoints    []string
	DefaultChannelBuffer int

	BaseDir          string
	TimeoutOptionSet bool
	DuplicateInput   int
	NumWarmUp        int

	DatasetPath      string
	DatasetImageUrls []string
	Count            int
}

func GetDefaultSettings

func GetDefaultSettings() (o Settings)