From ade5fb9b590ae637ba8bdf1796957da13c6ac279 Mon Sep 17 00:00:00 2001 From: Anshuman Mishra Date: Sat, 1 Jul 2023 15:41:30 +0530 Subject: [PATCH] fixed formatting --- guides/ipynb/keras_nlp/getting_started.ipynb | 6 ++---- guides/keras_nlp/getting_started.py | 4 +--- 2 files changed, 3 insertions(+), 7 deletions(-) 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,