training

package

v3.0.0 Latest Latest Go to latest Published: Oct 11, 2026 License: MIT Imports: 9 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

func NewAdam(lr, beta, beta2, epsilon float64) *Adam

func (o *Adam) Init(size int)

func (o *Adam) Update(value, gradient float64, t, idx int) float64

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

func NewBatchTrainer(solver Solver, verbosity, batchSize, parallelism int) *BatchTrainer

func (t *BatchTrainer) Train(n *ml.Neural, examples, validation Examples, iterations int)

type Example struct {
	Input    []float64
	Response []float64
}

type Examples []Example

func (e Examples) Shuffle()

func (e Examples) Split(p float64) (first, second Examples)

func (e Examples) SplitN(n int) []Examples

func (e Examples) SplitSize(size int) []Examples

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

func NewTrainer(solver Solver, verbosity int) *OnlineTrainer

func (t *OnlineTrainer) Train(n *ml.Neural, examples, validation Examples, iterations int)

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

func NewSGD(lr, momentum, decay float64, nesterov bool) *SGD

func (o *SGD) Init(size int)

func (o *SGD) Update(value, gradient float64, iteration, idx int) float64

type Solver interface {
	Init(size int)
	Update(value, gradient float64, interation, idx int) float64
}

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

func NewStatsPrinter() *StatsPrinter

func (p *StatsPrinter) Init(n *ml.Neural)

func (p *StatsPrinter) PrintProgress(n *ml.Neural, validation Examples, elapsed time.Duration, iteration int)

type Trainer interface {
	Train(n *ml.Neural, examples, validation Examples, iterations int)
}