Fixing up docs for make_parse_example_spec

PiperOrigin-RevId: 270068514
This commit is contained in:
Rohan Jain 2019-09-19 10:15:08 -07:00 committed by TensorFlower Gardener
parent 185f29aadb
commit d290833308

View File

@ -766,17 +766,18 @@ def make_parse_example_spec_v2(feature_columns):
```python ```python
# Define features and transformations # Define features and transformations
feature_a = categorical_column_with_vocabulary_file(...) feature_a = tf.feature_column.categorical_column_with_vocabulary_file(...)
feature_b = numeric_column(...) feature_b = tf.feature_column.numeric_column(...)
feature_c_bucketized = bucketized_column(numeric_column("feature_c"), ...) feature_c_bucketized = tf.feature_column.bucketized_column(
feature_a_x_feature_c = crossed_column( tf.feature_column.numeric_column("feature_c"), ...)
feature_a_x_feature_c = tf.feature_column.crossed_column(
columns=["feature_a", feature_c_bucketized], ...) columns=["feature_a", feature_c_bucketized], ...)
feature_columns = set( feature_columns = set(
[feature_b, feature_c_bucketized, feature_a_x_feature_c]) [feature_b, feature_c_bucketized, feature_a_x_feature_c])
features = tf.io.parse_example( features = tf.io.parse_example(
serialized=serialized_examples, serialized=serialized_examples,
features=make_parse_example_spec(feature_columns)) features=tf.feature_column.make_parse_example_spec(feature_columns))
``` ```
For the above example, make_parse_example_spec would return the dict: For the above example, make_parse_example_spec would return the dict: