Versions in this module Expand all Collapse all v1 v1.55.0 Jan 12, 2024 Changes in this version + const BatchSize + const Beta1 + const Beta2 + var LearningRate = 0.01 + func CalculateCostGradient(output, target float64) float64 + func Run(ctx context.Context, datasetProvider IDatasetProvider, ...) error + func ValidationCost(output, target float64) float64 + type Gradient struct + M1 float64 + M2 float64 + Value float64 + func (g *Gradient) Apply(elem *float64) + func (g *Gradient) Calculate() float64 + func (g *Gradient) Reset() + func (g *Gradient) Update(delta float64) + type IDatasetProvider interface + Load func(ctx context.Context, dataset chan<- domain.DatasetItem) error + type ITunableEvaluator interface + ComputeFeatures func(pos *common.Position) domain.TuneEntry + EnableTuning func() + StartingWeights func() []float64 + type Sample struct + Target float32 + type ThreadData struct + type Trainer struct + func NewTrainer(training, validation []Sample, startingWeights []float64, threads int, ...) *Trainer + func (t *Trainer) Sigmoid(x float64) float64 + func (t *Trainer) SigmoidPrime(x float64) float64 + func (t *Trainer) Train(epochs int) error