Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Network ¶
type Network struct {
// HiddenWeights []*mat.Dense
// OutputWeights *mat.Dense
HiddenWeights []mat.Matrix
OutputWeights mat.Matrix
HiddenUpdate []mat.Matrix
OutputUpdate mat.Matrix
HiddenV []mat.Matrix
OutputV mat.Matrix
HiddenMomentum []mat.Matrix
OutputMomentum mat.Matrix
SgdMomentum float64
LayerOut []mat.Matrix
// contains filtered or unexported fields
}
Network is a neural network with 3 layers
func CreateNetwork ¶
func CreateNetwork(input int, hidden []int, output int, rate float64, decay float64, SgdMomentum float64, beta1 float64, beta2 float64, eps float64) (net Network)
CreateNetwork creates a neural network with random weights
func (*Network) AdamUpdate ¶
Click to show internal directories.
Click to hide internal directories.