10 lines
353 B
Nim
10 lines
353 B
Nim
import nn/network
|
|
import nn/util/activations
|
|
import nn/util/losses
|
|
|
|
|
|
var net = newNeuralNetwork(@[2, 3, 2], activationFunc=newActivation(sigmoid, proc (x, y: float): float = 0.0),
|
|
lossFunc=newLoss(mse, mse), weightRange=(-1.0, +1.0), learnRate=0.05)
|
|
var prediction = net.predict(newMatrix[float](@[2.7, 3.0]))
|
|
echo prediction
|