From 93dcb1f4265407c8cbf322a0683eba4d4a5b483a Mon Sep 17 00:00:00 2001
From: Raghuram Subramani <raghus2247@gmail.com>
Date: Sun, 9 Jun 2024 10:18:00 +0530
Subject: NN

---
 src/loss.py | 5 +++++
 1 file changed, 5 insertions(+)
 create mode 100644 src/loss.py

(limited to 'src/loss.py')

diff --git a/src/loss.py b/src/loss.py
new file mode 100644
index 0000000..625ff0f
--- /dev/null
+++ b/src/loss.py
@@ -0,0 +1,5 @@
+from .scalar import Scalar
+
+def mse(ys, preds) -> Scalar:
+    losses = [ (y - pred) ** 2 for y, pred in zip(ys, preds) ]
+    return sum(losses)
-- 
cgit v1.2.3