diff --git a/04_mnist_basics.ipynb b/04_mnist_basics.ipynb index 908cf1927..00562c3b1 100644 --- a/04_mnist_basics.ipynb +++ b/04_mnist_basics.ipynb @@ -3833,7 +3833,7 @@ "cell_type": "markdown", "metadata": {}, "source": [ - "Let's check our accuracy. To decide if an output represents a 3 or a 7, we can just check whether it's greater than 0.5, so our accuracy for each item can be calculated (using broadcasting, so no loops!) with:" + "Let's check our accuracy. To decide if an output represents a 3 or a 7, we can just check whether it's greater than 0.0, so our accuracy for each item can be calculated (using broadcasting, so no loops!) with:" ] }, { @@ -3859,7 +3859,7 @@ } ], "source": [ - "corrects = (preds>0.5).float() == train_y\n", + "corrects = (preds>0.0).float() == train_y\n", "corrects" ] },