description: Retrieves a dict mapping words to their index in the IMDB dataset.

tf.keras.datasets.imdb.get_word_index

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.