utils

Documentation

Index

• func NewNetwork(Config models.NeuralNetConfig) *models.NeuralNet
• func Predict(x *mat.Dense, nn *models.NeuralNet) (*mat.Dense, error)
• func ShuffleRawCSVdata(data [][]string) [][]string
• func Sigmoid(x float64) float64
• func SigmoidPrime(x float64) float64
• func SplitData(data dataframe.DataFrame)
• func SumAlongAxis(axis int, m *mat.Dense) (*mat.Dense, error)
• func Train(x, y *mat.Dense, nn *models.NeuralNet) (*mat.Dense, error)

Functions

func NewNetwork

func NewNetwork(Config models.NeuralNetConfig) *models.NeuralNet

NewNetwork for obtaining new network

func Predict

func Predict(x *mat.Dense, nn *models.NeuralNet) (*mat.Dense, error)

Predict for prediction

func ShuffleRawCSVdata

func ShuffleRawCSVdata(data [][]string) [][]string

ShuffleRawCSVdata function for shuffling dataset

func Sigmoid

func Sigmoid(x float64) float64

Sigmoid Activation Function

func SigmoidPrime

func SigmoidPrime(x float64) float64

SigmoidPrime Derivative of Sigmoid Function

func SplitData

func SplitData(data dataframe.DataFrame)

SplitData function for spliting data set

func SumAlongAxis

func SumAlongAxis(axis int, m *mat.Dense) (*mat.Dense, error)

SumAlongAxis function for sum

func Train

func Train(x, y *mat.Dense, nn *models.NeuralNet) (*mat.Dense, error)

Train functions