loss.py « src - autograd.git - An implementation of autograd / backpropagation.

about summary refs log tree commit diff

path: root/src/loss.py

blob: 625ff0f0be0f23109c5ec51a44fc90feb388667e (plain)

from .scalar import Scalar

def mse(ys, preds) -> Scalar:
    losses = [ (y - pred) ** 2 for y, pred in zip(ys, preds) ]
    return sum(losses)