just an implementation of neural network model for imdb from Deep Learning With Python book

imdb_model.py

from keras import models
from keras import layers

def model(training_data, training_label, epochs=20):
    model = models.Sequential()
    model.add(layers.Dense(16, activation='relu', input_shape=(10000,)))
    model.add(layers.Dense(16, activation='relu'))
    model.add(layers.Dense(1, activation='sigmoid'))
    model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics=['accuracy'])
    model.fit(x_train, y_train, epochs=epochs, batch_size=512)
    return model
~