Skip to content

Instantly share code, notes, and snippets.

@kidtronnix

kidtronnix/predict.go

Created September 7, 2018 14:58
Show Gist options
  • Save kidtronnix/ffcb24f46edbbb90db13cd74359a7e38 to your computer and use it in GitHub Desktop.
Save kidtronnix/ffcb24f46edbbb90db13cd74359a7e38 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict.go
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]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment