Updated Large-scale multi-label text classification example for Keras v3

keras-team · Mar 2, 2024 · 9659c55 · 9659c55
1 parent a3f7572
commit 9659c55
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/examples/nlp/.DS_Store b/examples/nlp/.DS_Store
diff --git a/examples/nlp/multi_label_classification.py b/examples/nlp/multi_label_classification.py
@@ -30,8 +30,9 @@
 ## Imports
 """
 
-from tensorflow.keras import layers
-from tensorflow import keras
+import keras
+from keras import layers
+
 import tensorflow as tf
 
 from sklearn.model_selection import train_test_split