internal

PiperOrigin-RevId: 246656018
liujiantao007 · May 4, 2019 · 3191b90 · 3191b90
1 parent fabc828
commit 3191b90
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/...ensorflow_for_deep_learning/l05c04_exercise_flowers_with_data_augmentation_solution.ipynb b/...ensorflow_for_deep_learning/l05c04_exercise_flowers_with_data_augmentation_solution.ipynb
@@ -281,7 +281,8 @@
         "  img_path = os.path.join(base_dir, cl)\n",
         "  images = glob.glob(img_path + '/*.jpg')\n",
         "  print(\"{}: {} Images\".format(cl, len(images)))\n",
-        "  train, val = images[:round(len(images)*0.8)], images[round(len(images)*0.8):]\n",
+        "  num_train = int(round(len(images)*0.8))\n",
+        "  train, val = images[:num_train], images[num_train:]\n",
         "\n",
         "  for t in train:\n",
         "    if not os.path.exists(os.path.join(base_dir, 'train', cl)):\n",
@@ -294,6 +295,19 @@
         "    shutil.move(v, os.path.join(base_dir, 'val', cl))"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": 0,
+      "metadata": {
+        "colab": {},
+        "colab_type": "code",
+        "id": "yP85YhYol8ER"
+      },
+      "outputs": [],
+      "source": [
+        "round(len(images)*0.8)"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {