From 2d76b3c7861e8f5d20ba6b37faf486fec4c68c45 Mon Sep 17 00:00:00 2001 From: Luca Pizzini Date: Tue, 5 Mar 2024 19:37:02 +0100 Subject: [PATCH] Updated Large-scale multi-label text classification example for Keras v3 --- examples/nlp/multi_label_classification.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/nlp/multi_label_classification.py b/examples/nlp/multi_label_classification.py index 14c944b657..d2de991c6c 100644 --- 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 @@ -145,7 +146,7 @@ """ terms = tf.ragged.constant(train_df["terms"].values) -lookup = tf.keras.layers.StringLookup(output_mode="multi_hot") +lookup = layers.StringLookup(output_mode="multi_hot") lookup.adapt(terms) vocab = lookup.get_vocabulary()