arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix statefull RNN's validation set range

Browse Source
This commit is contained in:
Aurélien Geron
2022-09-12 11:48:12 +12:00
parent 7a551f5fb1
commit de0f184265
1 changed files with 2 additions and 1 deletions
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)"
] ]
}, },