diff --git a/src/main.nim b/src/main.nim index c316786..e4cd191 100644 --- a/src/main.nim +++ b/src/main.nim @@ -8,4 +8,3 @@ var mlp = newNeuralNetwork(@[newDenseLayer(2, 3, Sigmoid), lossFunc=MSE, learnRate=0.05, momentum=0.55, weightRange=(start: -1.0, stop: 1.0), biasRange=(start: -1.0, stop: 1.0)) echo mlp.feedforward(newMatrix[float](@[1.0, 2.0])) - diff --git a/src/nn/network.nim b/src/nn/network.nim index 47c2a38..336e87a 100644 --- a/src/nn/network.nim +++ b/src/nn/network.nim @@ -98,7 +98,7 @@ proc newDenseLayer*(inputSize: int, outputSize: int, activationFunc: Activation) proc newNeuralNetwork*(topology: seq[Layer], lossFunc: Loss, learnRate: float, momentum: float, weightRange, biasRange: tuple[start, stop: float]): NeuralNetwork = ## Initializes a new neural network with - ## the given topology and iperparameters. + ## the given topology and hyperparameters. ## Weights and biases are initialized with ## random values in the chosen range new(result)