13 lines
453 B
Nim
13 lines
453 B
Nim
import nn/network
|
|
import nn/util/matrix
|
|
|
|
|
|
var mlp = newNeuralNetwork(@[newDenseLayer(784, 10),
|
|
newDenseLayer(10, 16),
|
|
newDenseLayer(16, 10)],
|
|
lossFunc=MSE, activationFunc=Softmax,
|
|
learnRate=0.05, momentum=0.55,
|
|
weightRange=(start: -1.0, stop: 1.0),
|
|
biasRange=(start: -1.0, stop: 1.0))
|
|
|