The highest tagged major version is
Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Adam ¶
type Adam struct {
// contains filtered or unexported fields
}
Adam is an Adam solver
type BatchTrainer ¶
type BatchTrainer struct {
// contains filtered or unexported fields
}
BatchTrainer implements parallelized batch training
func NewBatchTrainer ¶
func NewBatchTrainer(solver Solver, verbosity, batchSize, parallelism int) *BatchTrainer
type Examples ¶
type Examples []Example
type OnlineTrainer ¶
type OnlineTrainer struct {
// contains filtered or unexported fields
}
OnlineTrainer is a basic, online network trainer
func NewTrainer ¶
func NewTrainer(solver Solver, verbosity int) *OnlineTrainer
type SGD ¶
type SGD struct {
// contains filtered or unexported fields
}
SGD is stochastic gradient descent with nesterov/momentum
type Solver ¶
type Solver interface {
Init(size int)
Update(value, gradient float64, interation, idx int) float64
}
Solver implements an update rule for training a Neural Network
type StatsPrinter ¶
type StatsPrinter struct {
// contains filtered or unexported fields
}
func NewStatsPrinter ¶
func NewStatsPrinter() *StatsPrinter
func (*StatsPrinter) Init ¶
func (p *StatsPrinter) Init(n *ml.Neural)
func (*StatsPrinter) PrintProgress ¶
Source Files
Click to show internal directories.
Click to hide internal directories.