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": [
|
||||
"train_set = quickdraw_dataset(train_files, shuffle_buffer_size=10000)\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