Skip to content

Instantly share code, notes, and snippets.

@adamoudad

adamoudad/load_imdb_dataset_keras.py

Created March 20, 2021 21:03
Show Gist options
  • Save adamoudad/e012e30edcd16090a2e8794ca5ff9044 to your computer and use it in GitHub Desktop.
Save adamoudad/e012e30edcd16090a2e8794ca5ff9044 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_imdb_dataset_keras.py
from tensorflow.keras.datasets import imdb
input_dim = 20000
(x_train, y_train), (x_val, y_val) = imdb.load_data(num_words=input_dim)
print(len(x_train), "Training sequences")
print(len(x_val), "Validation sequences")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment