| View source on GitHub |
Retrieves a dict mapping words to their index in the IMDB dataset.
tf.keras.datasets.imdb.get_word_index(
path='imdb_word_index.json'
)
| Arguments | |
|---|---|
path | where to cache the data (relative to ~/.keras/dataset). |
| Returns | |
|---|---|
| The word index dictionary. Keys are word strings, values are their index. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/datasets/imdb/get_word_index