sgd

Documentation

Index

• type Operation
• type Option
  • func WithBatchSize(batchSize int) Option
  • func WithLearningRate(learningRate float64) Option
• type SGD
  • func New(options ...Option) *SGD
  • func (o *SGD) Add(layer int, activation, cost *tensor.Vector)
  • func (o *SGD) Init(neurons *[][]int)
  • func (o *SGD) Optimizations() *[][][]float64

Types

type Operation

type Operation func(*SGD)

type Option

type Option func(*SGD)

func WithBatchSize

func WithBatchSize(batchSize int) Option

func WithLearningRate

func WithLearningRate(learningRate float64) Option

type SGD

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

func New

func New(options ...Option) *SGD

func (*SGD) Add

func (o *SGD) Add(layer int, activation, cost *tensor.Vector)

func (*SGD) Init

func (o *SGD) Init(neurons *[][]int)

func (*SGD) Optimizations

func (o *SGD) Optimizations() *[][][]float64