ann

package
v0.0.0-...-14a59ba Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Oct 27, 2017 License: MIT Imports: 6 Imported by: 0

README

Neural Network

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func RandomInitVector

func RandomInitVector(dim int64) *core.Vector

Types

type NeuralNetwork

type NeuralNetwork struct {
	Model    TwoLayerWeights
	MaxLabel int64
	Params   NeuralNetworkParams
}

Please refer to this chapter to know algorithm details : http://www4.rgu.ac.uk/files/chapter3%20-%20bp.pdf

func (*NeuralNetwork) Clear

func (algo *NeuralNetwork) Clear()

func (*NeuralNetwork) Command

func (algo *NeuralNetwork) Command() cli.Command

func (*NeuralNetwork) Evaluate

func (algo *NeuralNetwork) Evaluate(dataset *core.DataSet)

func (*NeuralNetwork) Init

func (algo *NeuralNetwork) Init(ctx *cli.Context)

func (*NeuralNetwork) LoadModel

func (self *NeuralNetwork) LoadModel(path string)

func (*NeuralNetwork) Predict

func (algo *NeuralNetwork) Predict(sample *core.Sample) float64

func (*NeuralNetwork) PredictMultiClass

func (algo *NeuralNetwork) PredictMultiClass(sample *core.Sample) *core.ArrayVector

func (*NeuralNetwork) SaveModel

func (self *NeuralNetwork) SaveModel(path string)

func (*NeuralNetwork) Train

func (algo *NeuralNetwork) Train(dataset *core.DataSet)

type NeuralNetworkParams

type NeuralNetworkParams struct {
	LearningRate         float64
	LearningRateDiscount float64
	Regularization       float64
	Hidden               int64
	Steps                int
	Verbose              int
}

type TwoLayerWeights

type TwoLayerWeights struct {
	L1 *core.Matrix
	L2 *core.Matrix
}

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL