Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tf.keras.datasets.imdb

TensorFlow 1 version

IMDB sentiment classification dataset.

Functions

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

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