adapt to tf 2.5.0 (#1564)

Co-authored-by: Haifeng Jin <[email protected]>
keras-team · May 16, 2021 · 09c06f7 · 09c06f7
1 parent 94ac836
commit 09c06f7
Show file tree

Hide file tree

Showing 5 changed files with 165 additions and 100 deletions.
diff --git a/autokeras/blocks/preprocessing.py b/autokeras/blocks/preprocessing.py
@@ -123,7 +123,10 @@ def build(self, hp, inputs=None):
         else:
             ngrams = hp.Int("ngrams", min_value=1, max_value=2, default=2)
         return preprocessing.TextVectorization(
-            max_tokens=self.max_tokens, ngrams=ngrams, output_mode="tf-idf"
+            max_tokens=self.max_tokens,
+            ngrams=ngrams,
+            output_mode="tf-idf",
+            pad_to_max_tokens=True,
         )(input_node)
 
     def get_config(self):