Documentation ¶
Index ¶
- func RandomInitVector(dim int64) *core.Vector
- type NeuralNetwork
- func (algo *NeuralNetwork) Evaluate(dataset *core.DataSet)
- func (algo *NeuralNetwork) Init(params map[string]string)
- func (self *NeuralNetwork) LoadModel(path string)
- func (algo *NeuralNetwork) Predict(sample *core.Sample) float64
- func (algo *NeuralNetwork) PredictMultiClass(sample *core.Sample) *core.ArrayVector
- func (self *NeuralNetwork) SaveModel(path string)
- func (algo *NeuralNetwork) Train(dataset *core.DataSet)
- type NeuralNetworkParams
- type TwoLayerWeights
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func RandomInitVector ¶
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) Evaluate ¶
func (algo *NeuralNetwork) Evaluate(dataset *core.DataSet)
func (*NeuralNetwork) Init ¶
func (algo *NeuralNetwork) Init(params map[string]string)
func (*NeuralNetwork) LoadModel ¶
func (self *NeuralNetwork) LoadModel(path string)
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 ¶
Click to show internal directories.
Click to hide internal directories.