![]() |
Class DenseFeatures
A layer that produces a dense Tensor
based on given feature_columns
.
Inherits From: DenseFeatures
Generally a single example in training data is described with FeatureColumns.
At the first layer of the model, this column oriented data should be converted
to a single Tensor
.
This layer can be called multiple times with different features.
This is the V2 version of this layer that uses name_scopes to create variables instead of variable_scopes. But this approach currently lacks support for partitioned variables. In that case, use the V1 version instead.
Example:
price = numeric_column('price')
keywords_embedded = embedding_column(
categorical_column_with_hash_bucket("keywords", 10K), dimensions=16)
columns = [price, keywords_embedded, ...]
feature_layer = DenseFeatures(columns)
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
dense_tensor = feature_layer(features)
for units in [128, 64, 32]:
dense_tensor = tf.keras.layers.Dense(units, activation='relu')(dense_tensor)
prediction = tf.keras.layers.Dense(1)(dense_tensor)
__init__
__init__(
feature_columns,
trainable=True,
name=None,
**kwargs
)
Creates a DenseFeatures object.
Args:
feature_columns
: An iterable containing the FeatureColumns to use as inputs to your model. All items should be instances of classes derived fromDenseColumn
such asnumeric_column
,embedding_column
,bucketized_column
,indicator_column
. If you have categorical features, you can wrap them with anembedding_column
orindicator_column
.trainable
: Boolean, whether the layer's variables will be updated via gradient descent during training.name
: Name to give to the DenseFeatures.**kwargs
: Keyword arguments to construct a layer.
Raises:
ValueError
: if an item infeature_columns
is not aDenseColumn
.