Merge pull request #13 from stefan-it/imdb-example-update

imdb example: fix deprecated warnings
bfelbo · Oct 17, 2017 · ac27e25 · ac27e25
2 parents 423f868 c5ce58a
commit ac27e25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/imdb_from_scratch.py b/examples/imdb_from_scratch.py
@@ -18,7  18,7 @@
 batch_size = 32
 
 print('Loading data...')
-(X_train, y_train), (X_test, y_test) = imdb.load_data(nb_words=nb_tokens)
 (X_train, y_train), (X_test, y_test) = imdb.load_data(num_words=nb_tokens)
 print(len(X_train), 'train sequences')
 print(len(X_test), 'test sequences')
 
@@ -37,7  37,7 @@
               metrics=['accuracy'])
 
 print('Train...')
-model.fit(X_train, y_train, batch_size=batch_size, nb_epoch=15,
 model.fit(X_train, y_train, batch_size=batch_size, epochs=15,
           validation_data=(X_test, y_test))
 score, acc = model.evaluate(X_test, y_test, batch_size=batch_size)
 print('Test score:', score)