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