change imdb batch_size

keras-team · May 31, 2021 · 8e11d2d · 8e11d2d
1 parent eea7a71
commit 8e11d2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/imdb.py b/examples/imdb.py
@@ -48,6 +48,6 @@ def imdb_raw():
 # Initialize the TextClassifier
 clf = ak.TextClassifier(max_trials=3)
 # Search for the best model.
-clf.fit(x_train, y_train, epochs=2)
+clf.fit(x_train, y_train, epochs=2, batch_size=8)
 # Evaluate on the testing data.
 print("Accuracy: {accuracy}".format(accuracy=clf.evaluate(x_test, y_test)))