04_mnist_basics: MSE lacks .sqrt() (#327)

2020-11-29 15:17:49 +01:00
parent 9455f8f058
commit 665c238467
1 changed files with 1 additions and 1 deletions
--- a/04_mnist_basics.ipynb
+++ b/04_mnist_basics.ipynb
@@ -3013,7 +3013,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "def mse(preds, targets): return ((preds-targets)**2).mean()"
+    "def mse(preds, targets): return ((preds-targets)**2).mean().sqrt()"
   ]
  },
  {