Skip to content

Instantly share code, notes, and snippets.

@Albertbol
Created March 12, 2021 06:39
Show Gist options
  • Save Albertbol/996e3937ddb129d927a8f96a0680d7c9 to your computer and use it in GitHub Desktop.
Save Albertbol/996e3937ddb129d927a8f96a0680d7c9 to your computer and use it in GitHub Desktop.
class NeuralNetwork {
constructor(
inputNodes,
hiddenNodes,
outputNodes,
learningRate,
bias,
trainingCycles,
normalizer
) {
this.inputNodes = inputNodes
this.hiddenNodes = hiddenNodes
this.outputNodes = outputNodes
this.learningRate = learningRate
this.weightsInputToHidden = Matrix.generate(
this.hiddenNodes,
this.inputNodes,
{ randomNormalized: true }
)
this.weightsHiddenToOutput = Matrix.generate(
this.outputNodes,
this.hiddenNodes,
{ randomNormalized: true }
)
this.bias = bias
this.biasForHidden = Matrix.generate(this.hiddenNodes, 1, {
staticNumber: this.bias,
})
this.biasForOutput = Matrix.generate(this.outputNodes, 1, {
staticNumber: this.bias,
})
this.trainingCycles = trainingCycles
this.normalizer = normalizer
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment