some changes

Codes/tfds.ipynb

@@ -376,7 +376,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 1,
+      "execution_count": null,
       "metadata": {
         "id": "Af8pDnDg1jSF"
       },
@@ -399,7 +399,7 @@
         "id": "tjNCb8xD18K5",
         "outputId": "890c2a8e-3820-4c5a-99b4-9149d878e99e"
       },
-      "execution_count": 2,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "execute_result",
@@ -453,7 +453,7 @@
         "id": "I-waHXUU2QGb",
         "outputId": "5016f908-7230-46d9-cc04-97002fa56cb6"
       },
-      "execution_count": 3,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -508,7 +508,7 @@
         "id": "8Bdzn3TM8SfU",
         "outputId": "f6df3027-a66c-4814-cf40-d9b0d6aa4d05"
       },
-      "execution_count": 4,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "execute_result",
@@ -790,7 +790,7 @@
         "id": "DbwDVgs42SIy",
         "outputId": "55d24cfa-e7ec-4b5e-d5ba-aa5342408603"
       },
-      "execution_count": 5,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "execute_result",
@@ -826,7 +826,7 @@
       "metadata": {
         "id": "VVDMZRr25BrH"
       },
-      "execution_count": 16,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -853,7 +853,7 @@
         "# Test Data\n",
         "# You don't need to call tf.data.Dataset.shuffle for test dataset.\n",
         "# Caching is done after batching, because batches can be the same between epochs.\n",
-        "ds_test = train_dataset.map(\n",
+        "ds_test = test_dataset.map(\n",
         "    preprocess_data, num_parallel_calls=tf.data.AUTOTUNE)\n",
         "ds_test = ds_test.batch(128)\n",
         "ds_test = ds_test.cache()\n",
@@ -862,7 +862,7 @@
       "metadata": {
         "id": "csxBEi2x5vJC"
       },
-      "execution_count": 18,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -890,7 +890,7 @@
         "id": "gTKuNLy_3MKL",
         "outputId": "4a44dbcf-f505-4ae1-ed56-89e4ca42a0e5"
       },
-      "execution_count": 24,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -918,7 +918,7 @@
         "id": "TAT1JlyNH7dP",
         "outputId": "f242b5cb-f251-4b56-f0b6-5f485187e04e"
       },
-      "execution_count": 26,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -954,7 +954,7 @@
       "metadata": {
         "id": "8yYHBMVO2kAc"
       },
-      "execution_count": 20,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -966,7 +966,7 @@
       "metadata": {
         "id": "tzCUNwef21CE"
       },
-      "execution_count": 21,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -984,7 +984,7 @@
         "id": "EWQ-Ywhq28DT",
         "outputId": "4587bf1d-9482-4c70-ee48-06f16a1e3863"
       },
-      "execution_count": 22,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",