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))