optimizers

package

v1.6.2 Latest Latest Go to latest Published: Jun 2, 2023 License: Apache-2.0 Imports: 4 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

This section is empty.

This section is empty.

This section is empty.

type AdaGrad struct {
	Epsilon float64
	// contains filtered or unexported fields
}

func (ada *AdaGrad) Initialize(n int)

func (ada *AdaGrad) Initialized() bool

func (ada *AdaGrad) Rescale(shift *mat.Dense, index int) *mat.Dense

func (ada *AdaGrad) Size() int

type Adam struct {
	Beta2   float64
	Beta1   float64
	Epsilon float64
	// contains filtered or unexported fields
}

func (adam *Adam) Initialize(n int)

func (adam *Adam) Initialized() bool

func (adam *Adam) Rescale(shift *mat.Dense, index int) *mat.Dense

func (adam *Adam) Size() int

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

func (g *GradientDescent) Initialize(n int)

func (g *GradientDescent) Initialized() bool

func (g *GradientDescent) Rescale(shifts *mat.Dense, _ int) *mat.Dense

func (g *GradientDescent) Size() int

type Momentum struct {
	Gamma float64
	// contains filtered or unexported fields
}

func (mom *Momentum) Initialize(n int)

func (mom *Momentum) Initialized() bool

func (mom *Momentum) Rescale(shift *mat.Dense, index int) *mat.Dense

func (mom *Momentum) Size() int

type Optimizer interface {
	Rescale(*mat.Dense, int) *mat.Dense

	Initialize(int)

	Size() int
	Initialized() bool
}

type RMSProp struct {
	Gamma   float64
	Epsilon float64
	// contains filtered or unexported fields
}

func (rms *RMSProp) Initialize(n int)

func (rms *RMSProp) Initialized() bool

func (rms *RMSProp) Rescale(shift *mat.Dense, index int) *mat.Dense

func (rms *RMSProp) Size() int