network

Documentation

Index

• type Network
  • func New(cowId int, numOfLayers int, sizesOfLayers []int, layerTypes []node.NodeType, ...) *Network
  • func (ne *Network) PredictClass(cowId int, examples []dataset.Example) (int, *Network)
  • func (ne *Network) ProcessInput(cowId int, examples []dataset.Example, iter int, rehash bool, rebuild bool) (float64, *Network)

Types

type Network

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

func New

func New(
	cowId int,
	numOfLayers int,
	sizesOfLayers []int,
	layerTypes []node.NodeType,
	batchSize int,
	learningRate float64,
	inputDim int,
	k []int,
	l []int,
	rangePow []int,
	sparsity []float64,
) *Network

func (*Network) PredictClass

func (ne *Network) PredictClass(
	cowId int,
	examples []dataset.Example,
) (int, *Network)

func (*Network) ProcessInput

func (ne *Network) ProcessInput(
	cowId int,
	examples []dataset.Example,
	iter int,
	rehash bool,
	rebuild bool,
) (float64, *Network)