Fix test_set should be based on eval_files[5:], not eval_files[:5], fixes #144
parent
397d2600f7
commit
d46228a6af
|
@ -1373,7 +1373,7 @@
|
||||||
"source": [
|
"source": [
|
||||||
"train_set = quickdraw_dataset(train_files, shuffle_buffer_size=10000)\n",
|
"train_set = quickdraw_dataset(train_files, shuffle_buffer_size=10000)\n",
|
||||||
"valid_set = quickdraw_dataset(eval_files[:5])\n",
|
"valid_set = quickdraw_dataset(eval_files[:5])\n",
|
||||||
"test_set = quickdraw_dataset(eval_files[:5])"
|
"test_set = quickdraw_dataset(eval_files[5:])"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
|
Loading…
Reference in New Issue