models

Documentation

Index

• type Model
  • func NewModel(argLayers interface{}, args ...interface{}) *Model
  • func (m *Model) Bprop(delta []backends.Tensor) []backends.Tensor
  • func (m *Model) ClassName() string
  • func (m *Model) Eval(dataset *data.DataIterator, metric transforms.Metric) []backends.Value
  • func (m *Model) Fit(dataset *data.DataIterator, cost layers.Cost, optimizer optimizers.Optimizer, ...)
  • func (m *Model) Fprop(x []backends.Tensor, inference bool) []backends.Tensor
  • func (m *Model) GetOutputs(dataset *data.DataIterator, writer data.DataWriter)
  • func (m *Model) Init(self base.Object, argLayers interface{}, args base.Args)
  • func (m *Model) Initialize(inObj []int, cost layers.Cost)
  • func (m *Model) LoadParams(paramPath string)
  • func (m *Model) SaveParams(paramPath string)
• type ParamReader
  • func NewParamReader(be generators.Generator, symbol string) *ParamReader
  • func (r *ParamReader) Read(x backends.Tensor)
• type ParamWriter
  • func NewParamWriter(be generators.Generator, symbol string) *ParamWriter
  • func (r *ParamWriter) Write(x backends.Tensor)

Types

type Model

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

func NewModel

func NewModel(argLayers interface{}, args ...interface{}) *Model

func (*Model) Bprop

func (m *Model) Bprop(delta []backends.Tensor) []backends.Tensor

func (*Model) ClassName

func (m *Model) ClassName() string

func (*Model) Eval

func (m *Model) Eval(dataset *data.DataIterator, metric transforms.Metric) []backends.Value

func (*Model) Fit

func (m *Model) Fit(
	dataset *data.DataIterator,
	cost layers.Cost,
	optimizer optimizers.Optimizer,
	numEpochs int,
	callbks *callbacks.Callbacks)

func (*Model) Fprop

func (m *Model) Fprop(x []backends.Tensor, inference bool) []backends.Tensor

func (*Model) GetOutputs

func (m *Model) GetOutputs(dataset *data.DataIterator, writer data.DataWriter)

func (*Model) Init

func (m *Model) Init(self base.Object, argLayers interface{}, args base.Args)

func (*Model) Initialize

func (m *Model) Initialize(inObj []int, cost layers.Cost)

func (*Model) LoadParams

func (m *Model) LoadParams(paramPath string)

func (*Model) SaveParams

func (m *Model) SaveParams(paramPath string)

type ParamReader

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

func NewParamReader

func NewParamReader(be generators.Generator, symbol string) *ParamReader

func (*ParamReader) Read

func (r *ParamReader) Read(x backends.Tensor)

type ParamWriter

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

func NewParamWriter

func NewParamWriter(be generators.Generator, symbol string) *ParamWriter

func (*ParamWriter) Write

func (r *ParamWriter) Write(x backends.Tensor)