mlengine

Documentation

Index

• type InterpreterOptions
• type MlEngine
  • func CreateMlEngine(cfg *config.Config) MlEngine
• type Tensor
  • func (t *Tensor) ByteSize() uint
  • func (t *Tensor) CopyFromBuffer(b interface{}) C.TfLiteStatus
  • func (t *Tensor) Float32s() []float32
  • func (t *Tensor) Type() TensorType
• type TensorType

Types

type InterpreterOptions

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

InterpreterOptions represents a TensorFlow Lite InterpreterOptions.

type MlEngine

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

func CreateMlEngine

func CreateMlEngine(cfg *config.Config) MlEngine

type Tensor

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

Tensor represents TensorFlow Lite Tensor.

func (*Tensor) ByteSize

func (t *Tensor) ByteSize() uint

ByteSize() returns byte size of the tensor.

func (*Tensor) CopyFromBuffer

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

CopyFromBuffer write buffer to the tensor.

func (*Tensor) Float32s

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

func (*Tensor) Type

func (t *Tensor) Type() TensorType

Type() returns TensorType.

type TensorType

type TensorType int

TensorType is types of the tensor.

const (
	TfLiteNoType TensorType = iota
	TfLiteFloat32
)

Tensors type