tf.keras.layers.DenseFeatures api as the candidate of hb.feature_column.DenseFeatures can not work with tf.feature_column.shared_embedding_columns about hybridbackend

[francktcheng]

We recommend you to use the tf.feature_column.input_layer instead of tf.keras.layers.DenseFeatures when the feature columns are generated by using tf.feature_column.shared_embedding_columns. Please refer to the code example as follows:

...
self._embedding_columns = tf.feature_column.shared_embedding_columns(
        [tf.feature_column.categorical_column_with_identity(
          key=f,
          num_buckets=self._args.data_spec.embedding_sizes[f],
          default_value=self._args.data_spec.defaults[f])
          for f in self._categorical_fields],
        dimension=self._args.embedding_dim,
        initializer=tf.random_uniform_initializer(-1e-3, 1e-3)
      )
cols_to_output_tensors = {}
      tf.feature_column.input_layer(
        {f: features[f] for f in self._categorical_fields},
        self._embedding_columns,
        cols_to_output_tensors=cols_to_output_tensors)
 ...

Internally, the tf.feature_column.input_layer API does accept feature columns with a type of _FeatureColumn while the API of tf.keras.layers.DenseFeatures only accepts a type derived from FeatureColumn which is not compatible with that of a shared embedding.