network

Documentation

Index

• type DataArray
• type Network
  • func Load(path string) (nt *Network, err error)
  • func NewNetwork(inputSize int, outputSize int, activation activation.IActivation) (nt *Network, err error)
  • func (nt *Network) AddLayer(size int, activation activation.IActivation) error
  • func (nt *Network) Save(path string) (err error)
  • func (nt *Network) SetProps(props Props)
  • func (nt *Network) Test(input DataArray) (DataArray, []int, error)
  • func (nt *Network) Train(inputs []DataArray, targets []DataArray) error
• type Props

Types

type DataArray

type DataArray []float64

type Network

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

func Load

func Load(path string) (nt *Network, err error)

func NewNetwork

func NewNetwork(inputSize int, outputSize int, activation activation.IActivation) (nt *Network, err error)

func (*Network) AddLayer

func (nt *Network) AddLayer(size int, activation activation.IActivation) error

AddLayer - Add single hidden layer

func (*Network) Save

func (nt *Network) Save(path string) (err error)

func (*Network) SetProps

func (nt *Network) SetProps(props Props)

SetProps - Self defined

func (*Network) Test

func (nt *Network) Test(input DataArray) (DataArray, []int, error)

Test neural network

func (*Network) Train

func (nt *Network) Train(inputs []DataArray, targets []DataArray) error

Train - Train neural network

type Props

type Props struct {
	Loss      loss.ILoss
	Optimizer optimizer.IOptimizer
	ErrLimit  float64
	MaxEpoch  int
}

Network