mlengine

package

v0.4.1 Latest Latest Go to latest Published: Feb 3, 2022 License: MIT Imports: 18 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.

This section is empty.

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

InterpreterOptions represents a TensorFlow Lite InterpreterOptions.

type MlEngine interface {
	GetSpamValue(issue *spamcheck.Issue) float32
}

func CreateMlEngine(cfg *config.Config) MlEngine

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

Tensor represents TensorFlow Lite Tensor.

func (t *Tensor) ByteSize() uint

ByteSize() returns byte size of the tensor.

func (t *Tensor) CopyFromBuffer(b interface{}) C.TfLiteStatus

CopyFromBuffer write buffer to the tensor.

func (t *Tensor) Float32s() []float32

func (t *Tensor) Type() TensorType

Type() returns TensorType.

type TensorType int

TensorType is types of the tensor.

const (
	TfLiteNoType TensorType = iota
	TfLiteFloat32
)

Tensors type