Announcing the TensorFlow Dev Summit 2020 Learn more

Module: tf.keras.datasets.imdb

TensorFlow 2 version

IMDB sentiment classification dataset.

Aliases:

Functions

get_word_index(...): Retrieves the dictionary mapping word indices back to words.

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