Added MSE message (#500)

* Added MSE message * changed wrong line. * Update examples/python/linear_regression.py Co-authored-by: Awni Hannun <[email protected]> --------- Co-authored-by: Awni Hannun <[email protected]>
2024-01-19 06:27:50 -08:00
parent f6feb61f92
commit 6589c869d6
1 changed files with 1 additions and 1 deletions
@@ -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)"
 )