From 93dcb1f4265407c8cbf322a0683eba4d4a5b483a Mon Sep 17 00:00:00 2001 From: Raghuram Subramani Date: Sun, 9 Jun 2024 10:18:00 +0530 Subject: NN --- neural_network.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) (limited to 'neural_network.py') diff --git a/neural_network.py b/neural_network.py index 62f3f9c..593a6ce 100755 --- a/neural_network.py +++ b/neural_network.py @@ -1,17 +1,29 @@ #!/usr/bin/env python from src.nn import MLP +from src.loss import mse X = [ - [ 0.0, 0.0, 0.0 ], - [ 1.0, 1.0, 1.0 ], + [ 0.0, 1.0, 2.0 ], + [ 2.0, 1.0, 0.0 ], [ 2.0, 2.0, 2.0 ], [ 3.0, 3.0, 3.0 ] ] -y = [ 1.0, 2.0, 3.0, 4.0 ] # X + 1 - +y = [ 1.0, -1.0, 1.0, -1.0 ] n = MLP(3, [ 4, 4, 1 ]) pred = [ n(x) for x in X ] print(pred) + +for i in range(400): + pred = [ n(x) for x in X ] + loss = mse(y, pred) + + loss.zero_grad() + loss.backward() + n.optimise(0.01) + + print(loss.data) + +print(pred) -- cgit v1.2.3