diff --git a/src/main.nim b/src/main.nim index 1d2d039..c316786 100644 --- a/src/main.nim +++ b/src/main.nim @@ -6,6 +6,6 @@ var mlp = newNeuralNetwork(@[newDenseLayer(2, 3, Sigmoid), newDenseLayer(3, 2, Sigmoid), newDenseLayer(2, 3, Softmax)], lossFunc=MSE, learnRate=0.05, momentum=0.55, - weightRange=(start: -1.0, stop: 1.0), biasRange=(start: -10.0, stop: 10.0)) + weightRange=(start: -1.0, stop: 1.0), biasRange=(start: -1.0, stop: 1.0)) echo mlp.feedforward(newMatrix[float](@[1.0, 2.0]))