optimizer

Documentation

Index

• type IOptimizer
  • func Generate(attr *profile.Attr) (optimizer IOptimizer, err error)
• type SGD
  • func NewSGD() *SGD
  • func (o *SGD) CalculateDelta(grad float64) float64
  • func (o *SGD) CallMe() string

Types

type IOptimizer

type IOptimizer interface {
	CalculateDelta(grad float64) float64
	CallMe() string
}

func Generate

func Generate(attr *profile.Attr) (optimizer IOptimizer, err error)

type SGD

type SGD struct {
	Alpha      float64
	M          float64
	IsNesterov bool
	V          float64
}

func NewSGD

func NewSGD() *SGD

func (*SGD) CalculateDelta

func (o *SGD) CalculateDelta(grad float64) float64

func (*SGD) CallMe

func (o *SGD) CallMe() string