12 lines
447 B
Nim
12 lines
447 B
Nim
import nn/network
|
|
import nn/util/matrix
|
|
|
|
|
|
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: -1.0, stop: 1.0))
|
|
echo mlp.feedforward(newMatrix[float](@[1.0, 2.0]))
|
|
|