NA

PiperOrigin-RevId: 616934117
tensorflow · Mar 18, 2024 · 42e3607 · 42e3607
1 parent 8f2dc52
commit 42e3607
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fairness_indicators/example_model.py b/fairness_indicators/example_model.py
@@ -71,7 +71,7 @@ def parse_function(serialized):
         filenames=[train_tf_file]).map(parse_function).batch(512)
     return train_dataset
 
-  text_embedding_column = hub.text_embedding_column(
+  text_embedding_column = hub.feature_column.text_embedding_column(
       key=text_feature, module_spec=module_spec)
 
   classifier = tf_estimator.DNNClassifier(