Remove assert

PiperOrigin-RevId: 484086748

Author: MarkDaoust

site/en/tutorials/images/classification.ipynb

@@ -1110,16 +1110,43 @@
       "outputs": [],
       "source": [
         "predictions_lite = classify_lite(sequential_1_input=img_array)['outputs']\n",
-        "score_lite = tf.nn.softmax(predictions_lite)\n",
-        "\n",
-        "assert np.allclose(predictions, predictions_lite)\n",
-        "\n",
+        "score_lite = tf.nn.softmax(predictions_lite)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "ZKP_GFeKUWb5"
+      },
+      "outputs": [],
+      "source": [
         "print(\n",
         "    \"This image most likely belongs to {} with a {:.2f} percent confidence.\"\n",
         "    .format(class_names[np.argmax(score_lite)], 100 * np.max(score_lite))\n",
         ")"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "Poz_iYgeUg_U"
+      },
+      "source": [
+        "The prediction generated by the lite model should be almost identical to the predictions generated by the original model:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "InXXDJL8UYC1"
+      },
+      "outputs": [],
+      "source": [
+        "print(np.max(np.abs(predictions - predictions_lite)))"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -1148,6 +1175,7 @@
     "colab": {
       "collapsed_sections": [],
       "name": "classification.ipynb",
+      "provenance": [],
       "toc_visible": true
     },
     "kernelspec": {