Versions in this module Expand all Collapse all v1 v1.6.1 Aug 14, 2026 Changes in this version + func WithSeed(seed uint64) func(*neuralModel) + func WithShuffle(shuffle bool) func(*fitConfig) v1.6.0 Aug 1, 2026 Changes in this version + func WithBatchSize(batchSize int) func(*neuralModel) + func WithDropout(dropout float64) func(*neuralModel) + func WithL2Regularization(lambd float64) func(*neuralModel) + func WithLogInterval(interval int) func(*fitConfig) + func WithVerbose(verbose bool) func(*fitConfig) + type NeuralConfig struct + Activation nncore.ActivationType + Mode nncore.ModeType + NNStructure []int + type NeuralModel interface + Evaluate func(x *mat.Dense, y *mat.Dense) float64 + Fit func(xTrain *mat.Dense, yTrain *mat.Dense, options ...func(*fitConfig)) []float64 + Predict func(x *mat.Dense) *mat.Dense + Save func(path string) + Summary func() + func Load(path string) NeuralModel + type NeuralNetwork interface + NewTrainer func(config TrainerConfig, options ...func(*neuralModel)) NeuralModel + func NewNeuralNetwork(config NeuralConfig) NeuralNetwork + type TrainerConfig struct + Epochs int + LearningRate float64 + Optimizer nncore.OptimizerType
Go to main page