tf.keras.datasets.imdb.get_word_index
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' )
Args | |
---|---|
path |
where to cache the data (relative to ~/.keras/dataset ). |
Returns | |
---|---|
The word index dictionary. Keys are word strings, values are their index. |
Example:
# Retrieve the training sequences. (x_train, _), _ = keras.datasets.imdb.load_data() # Retrieve the word index file mapping words to indices word_index = keras.datasets.imdb.get_word_index() # Reverse the word index to obtain a dict mapping indices to words inverted_word_index = dict((i, word) for (word, i) in word_index.items()) # Decode the first sequence in the dataset decoded_sequence = " ".join(inverted_word_index[i] for i in x_train[0])
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/datasets/imdb/get_word_index