TensorFlow 1 version | 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. |