From 6589c869d6f2c0ba7c95fec50adfaf146f35d0b0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Arda=20Or=C3=A7un?= <109769982+Stealeristaken@users.noreply.github.com> Date: Fri, 19 Jan 2024 17:27:50 +0300 Subject: [PATCH] Added MSE message (#500) * Added MSE message * changed wrong line. * Update examples/python/linear_regression.py Co-authored-by: Awni Hannun --------- Co-authored-by: Awni Hannun --- examples/python/linear_regression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/python/linear_regression.py b/examples/python/linear_regression.py index 1de4fc531..c247d1701 100644 --- a/examples/python/linear_regression.py +++ b/examples/python/linear_regression.py @@ -41,6 +41,6 @@ error_norm = mx.sum(mx.square(w - w_star)).item() ** 0.5 throughput = num_iters / (toc - tic) print( - f"Loss {loss.item():.5f}, |w-w*| = {error_norm:.5f}, " + f"Loss {loss.item():.5f}, L2 distance: |w-w*| = {error_norm:.5f}, " f"Throughput {throughput:.5f} (it/s)" )