NA

PiperOrigin-RevId: 616934117

Author: zhouhao138

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(