Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
View Source
var ClassNames = []string{"CLASS 1", "CLASS 2", "CLASS 3"}
ClassNames to storeClass Names
Functions ¶
This section is empty.
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
Click to show internal directories.
Click to hide internal directories.