def testMakeParseExampleSpec(self):
text_column = hub.sparse_text_embedding_column("text",
self.spec,
combiner="mean",
default_value=None,
trainable=False)
parsing_spec = tf_v1.feature_column.make_parse_example_spec(
[text_column])
self.assertEqual(parsing_spec,
{"text": tf_v1.VarLenFeature(tf.string)})
def parse_example_spec(self):
"""Returns a `tf.Example` parsing spec as dict."""
return {self.key: tf_v1.VarLenFeature(tf.string)}