Skip to content

Instantly share code, notes, and snippets.

View kidtronnix's full-sized avatar

Simon Maxwell-Stewart kidtronnix

View GitHub Profile
package nn
import "gonum.org/v1/gonum/mat"
func (n *MLP) Predict(x mat.Matrix) mat.Matrix {
as, _ := n.forward(x)
return as[len(as)-1]
}
package nn
import (
"gonum.org/v1/gonum/mat"
)
func (n *MLP) Evaluate(x, y mat.Matrix) float64 {
p := n.Predict(x)
N, _ := p.Dims()
package nn
import gonum.org/v1/gonum/mat
type NN interface {
Train(x,y mat.Matrix)
Evaluate(x, y mat.Matrix) float64
Predict(x mat.Matrix) mat.Matrix
}