stack

Documentation

Index

• type Model
  • func Make(size int, callback func(i int) nn.StandardModel) *Model
  • func New(layers ...nn.StandardModel) *Model
  • func (m *Model) Forward(xs ...ag.Node) []ag.Node
  • func (m *Model) LastLayer() nn.Model

Types

type Model

type Model struct {
	nn.BaseModel
	Layers []nn.StandardModel
}

Model contains the sub-models.

func Make

func Make(size int, callback func(i int) nn.StandardModel) *Model

Make makes a new model by obtaining each layer with a callback.

func New

func New(layers ...nn.StandardModel) *Model

New returns a new model.

func (*Model) Forward

func (m *Model) Forward(xs ...ag.Node) []ag.Node

Forward performs the forward step for each input node and returns the result.

func (*Model) LastLayer

func (m *Model) LastLayer() nn.Model

LastLayer returns the last layer from the stack.