TensorFlow API r1.11 Python Module: tf.keras.datasets.imdb Defined in tensorflow/keras/datasets/imdb/__init__.py. IMDB sentiment classification dataset. Functions get_word_index(...): Retrieves the dictionary mapping word indices back to words. load_data(...): Loads the IMDB dataset. Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 3.0 License, and code samples are licensed under the Apache 2.0 License. For details, see our Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Last updated September 27, 2018.