diff --git a/guides/ipynb/keras_nlp/getting_started.ipynb b/guides/ipynb/keras_nlp/getting_started.ipynb index 42a645ad31..3ae5fa8b81 100644 --- a/guides/ipynb/keras_nlp/getting_started.ipynb +++ b/guides/ipynb/keras_nlp/getting_started.ipynb @@ -184,7 +184,7 @@ "\n", "# Inspect first review\n", "# Format is (review text tensor, label tensor)\n", - "print(imdb_train.unbatch().take(1).get_single_element())\n" + "print(imdb_train.unbatch().take(1).get_single_element())" ] }, { @@ -385,9 +385,7 @@ ")\n", "\n", "classifier = keras_nlp.models.BertClassifier.from_preset(\n", - " \"bert_tiny_en_uncased\",\n", - " preprocessor=None,\n", - " num_classes=2\n", + " \"bert_tiny_en_uncased\", preprocessor=None, num_classes=2\n", ")\n", "classifier.fit(\n", " imdb_train_cached,\n", diff --git a/guides/keras_nlp/getting_started.py b/guides/keras_nlp/getting_started.py index 66f7af39a7..b75cb85e6e 100644 --- a/guides/keras_nlp/getting_started.py +++ b/guides/keras_nlp/getting_started.py @@ -250,9 +250,7 @@ ) classifier = keras_nlp.models.BertClassifier.from_preset( - "bert_tiny_en_uncased", - preprocessor=None, - num_classes=2 + "bert_tiny_en_uncased", preprocessor=None, num_classes=2 ) classifier.fit( imdb_train_cached,