tf

package

v0.0.6 Latest Latest Go to latest Published: Dec 8, 2021 License: MIT Imports: 3 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

This section is empty.

This section is empty.

func Version() string

Version tensorflow version

type Model struct {
	// contains filtered or unexported fields
}

func New(name, exportDir string, tags, inputParamKey []string, outputParamKey string) *Model

New according the input params to generate the special tf model

func Register(name, exportDir string, tags []string) (*Model, error)

Register register and load model

func RegisterWithParamName(name, exportDir string, tags, inputParamKey []string, outputParamKey string) (*Model, error)

RegisterWithParamName register with param key, and load model

func (m *Model) Destruct() error

Destruct destroy model

func (m *Model) Load() error

Load tf model from special path

func (m *Model) Name() string

func (m *Model) Path() string

func (m *Model) Predict(dataSet []interface{}) (ret interface{}, err error)

Predict tf predict