diff options
author | Raghuram Subramani <raghus2247@gmail.com> | 2024-06-09 08:50:49 +0530 |
---|---|---|
committer | Raghuram Subramani <raghus2247@gmail.com> | 2024-06-09 08:50:49 +0530 |
commit | 194f7d40561485f5ac3a3556721cfbc542be3b07 (patch) | |
tree | 49edbcc9f1daf378396a8dd6a31c0078df3976f6 /gradient_descent.py | |
parent | 40240b0b383abc2d3e81e2bcfe5e4b6d6fdfec2a (diff) |
Update
Diffstat (limited to 'gradient_descent.py')
-rwxr-xr-x | gradient_descent.py | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/gradient_descent.py b/gradient_descent.py new file mode 100755 index 0000000..3f7f124 --- /dev/null +++ b/gradient_descent.py @@ -0,0 +1,37 @@ +#!/usr/bin/env python + +from src.scalar import Scalar +from src.graph import Graph + +x1 = Scalar(2, label='x1') +x2 = Scalar(0, label='x2') + +w1 = Scalar(-3, label='w1') +w2 = Scalar(1, label='w2') + +b = Scalar(6.7, label='b') + +x1w1 = x1 * w1; x1w1.label = 'x1w1' +x2w2 = x2 * w2; x2w2.label = 'x2w2' + +x1w1x2w2 = x1w1 + x2w2; x1w1x2w2.label = 'x1w1 + x2w2' + +L = x1w1x2w2 + b; L.label = 'L' + +o = L.tanh(); o.label = 'o' + +o.zero_grad() +o.backward() + +Graph(o).show() + +e = 2 * L +f = e.exp() +a = f - 1 +b = f + 1 +o = a / b + +o.zero_grad() +o.backward() + +Graph(o).show() |