ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf.keras.datasets.imdb

Public API for tf.keras.datasets.imdb namespace.

Functions

get_word_index(...): Retrieves a dict mapping words to their index in the IMDB dataset.

load_data(...): Loads the IMDB dataset.