Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Framework ¶
type Framework interface {
Load(modelPath string, modelFile string, model *Model) (err error)
Run(model *Model) (out map[string]interface{}, err error)
FrameworkTyp() string
}
Framework interface used to implement specific ml framework implementations
type Metadata ¶
type Metadata struct {
Name string
Inputs struct {
Params map[string]OperationParam
Features map[string]Feature
}
Outputs map[string]OperationParam
Method string
}
type Model ¶
type Model struct {
Metadata *Metadata
Instance interface{}
Inputs map[string]map[string]interface{}
}
Model represents a ML model. The metadata defines inputs, outputs, shapes, etc. The Instance object is a pointer to the actual framework obj. This will be used by the framework impl to execute models
func (*Model) AppendInput ¶
func (*Model) RemoveInput ¶
type OperationParam ¶
Click to show internal directories.
Click to hide internal directories.