models

Documentation

Index

• Variables
• type ModelConfig
• type ModelStats
• type NeuralNet
• type NeuralNetConfig

Variables

var ClassNames = []string{"CLASS 1", "CLASS 2", "CLASS 3"}

ClassNames to storeClass Names

Types

type ModelConfig

type ModelConfig struct {
	InputNeurons  int         `json:"inputNeurons"`
	OutputNeurons int         `json:"outputNeurons"`
	HiddenNeurons int         `json:"hiddenNeurons"`
	NumEpochs     int         `json:"numEpochs"`
	LearningRate  float64     `json:"learningRate"`
	BHidden       [][]float64 `json:"bHidden"`
	BHiddenDims   []int       `json:"bHiddenDims"`
	WHidden       [][]float64 `json:"wHidden"`
	WHiddenDims   []int       `json:"wHiddenDims"`
	BOut          [][]float64 `json:"bOut"`
	BOutDims      []int       `json:"bOutDims"`
	WOut          [][]float64 `json:"wOut"`
	WOutDims      []int       `json:"wOutDims"`
}

ModelConfig structure for saving model details

type ModelStats

type ModelStats struct {
	TestAccuracy    float64 `json:"testAccuracy"`
	TrainAccuracy   float64 `json:"trainAccuracy"`
	PredictAccuracy float64 `json:"predictAccuracy"`
}

ModelStats structure for model statistics

type NeuralNet

type NeuralNet struct {
	Config  NeuralNetConfig
	WHidden *mat.Dense
	BHidden *mat.Dense
	WOut    *mat.Dense
	BOut    *mat.Dense
}

NeuralNet structure

type NeuralNetConfig

type NeuralNetConfig struct {
	InputNeurons  int
	OutputNeurons int
	HiddenNeurons int
	NumEpochs     int
	LearningRate  float64
}

NeuralNetConfig structure