Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tf.compat.v2.keras.datasets.imdb

View source on GitHub

IMDB sentiment classification dataset.

Functions

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

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