experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change dataset to _dataset in model_maker colab.

Browse Source 
PiperOrigin-RevId: 350460252
Change-Id: I380169371cec5f4608f1ea865f95b2446435bbc3
This commit is contained in:
Yuqi Li 2021-01-06 17:37:24 -08:00 committed by TensorFlower Gardener
parent 2491d011f1
commit c96dba97d8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb
View File

@ -592,7 +592,7 @@
"outputs": [],
"source": [
"plt.figure(figsize=(10,10))\n",
"for i, (image, label) in enumerate(data.dataset.take(25)):\n",
"for i, (image, label) in enumerate(data.gen_dataset().unbatch().take(25)):\n",
" plt.subplot(5,5,i+1)\n",
" plt.xticks([])\n",
" plt.yticks([])\n",
@ -696,7 +696,7 @@
"# dataset, we will highlight it in red color.\n",
"plt.figure(figsize=(20, 20))\n",
"predicts = model.predict_top_k(test_data)\n",
"for i, (image, label) in enumerate(test_data.dataset.take(100)):\n",
"for i, (image, label) in enumerate(test_data.gen_dataset().unbatch().take(100)):\n",
" ax = plt.subplot(10, 10, i+1)\n",
" plt.xticks([])\n",
" plt.yticks([])\n",