feedforward

Documentation

Index

• type Layer
• type Network
  • func Load(src io.Reader, opt ...NetworkOpt) (*Network, error)
  • func New(opt ...NetworkOpt) *Network
  • func (nn *Network) EpochStats(epoch int) stats.Epoch
  • func (nn *Network) Predict(input []float64) []float64
  • func (nn *Network) Save(w io.Writer) error
  • func (nn *Network) String() string
  • func (nn *Network) Train(epochs int, inputs, targets [][]float64, callback func(int) bool)
  • func (nn *Network) TrainingDuration() time.Duration
• type NetworkOpt
  • func Activation(v func(float64) float64) NetworkOpt
  • func ActivationDerivative(v func(float64) float64) NetworkOpt
  • func LearningRate(v float64) NetworkOpt
  • func Shapes(v []int) NetworkOpt
• type Node

Types

type Layer

type Layer struct {
	Nodes []*Node
}

type Network

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

func Load

func Load(src io.Reader, opt ...NetworkOpt) (*Network, error)

func New

func New(opt ...NetworkOpt) *Network

func (*Network) EpochStats

func (nn *Network) EpochStats(epoch int) stats.Epoch

func (*Network) Predict

func (nn *Network) Predict(input []float64) []float64

func (*Network) Save

func (nn *Network) Save(w io.Writer) error

func (*Network) String

func (nn *Network) String() string

func (*Network) Train

func (nn *Network) Train(epochs int, inputs, targets [][]float64, callback func(int) bool)

func (*Network) TrainingDuration

func (nn *Network) TrainingDuration() time.Duration

type NetworkOpt

type NetworkOpt func(*networkOpts)

func Activation

func Activation(v func(float64) float64) NetworkOpt

func ActivationDerivative

func ActivationDerivative(v func(float64) float64) NetworkOpt

func LearningRate

func LearningRate(v float64) NetworkOpt

func Shapes

func Shapes(v []int) NetworkOpt

type Node

type Node struct {
	Weights []float64
	Bias    float64
}