nn

package

v1.0.3 Latest Latest Go to latest Published: Jun 5, 2025 License: GPL-3.0 Imports: 6 Imported by: 0

Details

View Source

const (
	ActivationReLu    = "relu"
	ActivationSoftmax = "softmax"
)

View Source

var ErrEmptyLayers = errors.New("empty layers")

func Solve(m *NeuralNetwork, x *gorgonia.Node, y *gorgonia.Node) (gorgonia.Value, error)

type Layer struct {
	Activation string
	Size       [2]int
	Dropout    float64
}

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

func NewNeuralNetwork(g *gorgonia.ExprGraph, layers []Layer) (*NeuralNetwork, error)

func (m *NeuralNetwork) Forward(x *gorgonia.Node) (err error)

func (m *NeuralNetwork) GetGraph() *gorgonia.ExprGraph

func (m *NeuralNetwork) GetVal() gorgonia.Value

func (m *NeuralNetwork) Learn(y *gorgonia.Node) error

func (m *NeuralNetwork) Learnables() gorgonia.Nodes