def load_imdb():
from keras.preprocessing import sequence<EOL>from keras.datasets import imdb<EOL>(X_train, y_train), (X_test, y_test) = imdb.load_data(nb_words=<NUM_LIT>)<EOL>X_train = sequence.pad_sequences(X_train, maxlen=<NUM_LIT:100>)<EOL>X_test = sequence.pad_sequences(X_test, maxlen=<NUM_LIT:100>)<EOL>return X_train, y_train, X_test, y_test<EOL>
Load IMDB dataset
Transform input data into an RDD of Sample