schuetoliver
/
handson-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix statefull RNN's validation set range

main
Aurélien Geron 2022-09-12 11:48:12 +12:00
parent 7a551f5fb1
commit de0f184265
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
16_nlp_with_rnns_and_attention.ipynb
View File

@ -735,7 +735,8 @@
" return ds.map(lambda window: (window[:, :-1], window[:, 1:])).prefetch(1)\n", " return ds.map(lambda window: (window[:, :-1], window[:, 1:])).prefetch(1)\n",
"\n", "\n",
"stateful_train_set = to_dataset_for_stateful_rnn(encoded[:1_000_000], length)\n", "stateful_train_set = to_dataset_for_stateful_rnn(encoded[:1_000_000], length)\n",
"stateful_valid_set = to_dataset_for_stateful_rnn(encoded[:1_000_000], length)\n", "stateful_valid_set = to_dataset_for_stateful_rnn(encoded[1_000_000:1_060_000],\n",
" length)\n",
"stateful_test_set = to_dataset_for_stateful_rnn(encoded[1_060_000:], length)" "stateful_test_set = to_dataset_for_stateful_rnn(encoded[1_060_000:], length)"
] ]
}, },