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)}