ai_models

Documentation

Index

• type NeuralNetwork
  • func NewNeuralNetwork(numInputs, numHiddenLayers, numNeuronsPerHiddenLayer, numOutputs int) *NeuralNetwork
  • func (nn *NeuralNetwork) Predict(x mat.Matrix) mat.Matrix
  • func (nn *NeuralNetwork) Train(X, y mat.Matrix) error

Types

type NeuralNetwork

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

NeuralNetwork represents a neural network model

func NewNeuralNetwork

func NewNeuralNetwork(numInputs, numHiddenLayers, numNeuronsPerHiddenLayer, numOutputs int) *NeuralNetwork

NewNeuralNetwork creates a new instance of the neural network

func (*NeuralNetwork) Predict

func (nn *NeuralNetwork) Predict(x mat.Matrix) mat.Matrix

Predict makes a prediction using the trained neural network

func (*NeuralNetwork) Train

func (nn *NeuralNetwork) Train(X, y mat.Matrix) error

Train trains the neural network using the provided training data