Function loss

05_2_gradientdescent_auto.py:19–20

(y, y_pred)

# loss = MSE
def loss(y, y_pred):
    return ((y_pred - y)**2).mean()

print(f'Prediction before training: f(5) = {forward(5).item():.3f}')