optimizers

Documentation

Index

• type OptimizationStrategy
• type Optimizer
  • func New(parameters nn.ParamChannelFunc, strategy OptimizationStrategy) *Optimizer
  • func (o *Optimizer) Optimize() error

Types

type OptimizationStrategy

type OptimizationStrategy interface {
	OptimizeParams(*nn.Param) error
}

OptimizationStrategy is the interface implemented by AdaGrad, Adam, etc.

type Optimizer

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

Optimizer is an optimizer that can optimize a set of parameters.

func New

func New(parameters nn.ParamChannelFunc, strategy OptimizationStrategy) *Optimizer

New returns a new optimizer.

func (*Optimizer) Optimize

func (o *Optimizer) Optimize() error

Optimize performs the optimization of the parameters.