tf.feature_column.indicator_column
| Просмотреть исходный код на GitHub |
Представляет многомерное представление заданного категориального столбца.
tf.feature_column.indicator_column(
categorical_column
)
Для модели DNN,
indicator_columnможно использовать для обертывания любогоcategorical_column_*(например, для подачи в DNN). Рассмотрите использованиеembedding_column, если количество ведер/уникальных значений велико.Для широкой (также линейной) модели,
indicator_columnявляется внутренним представлением категориального столбца при непосредственном передаче категориального столбца (как любого элемента в feature_columns) вlinear_model. Подробности см. вlinear_model.
name = indicator_column(categorical_column_with_vocabulary_list(
'name', ['bob', 'george', 'wanda']))
columns = [name, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
dense_tensor = input_layer(features, columns)
dense_tensor == [[1, 0, 0]] # If "name" bytes_list is ["bob"]
dense_tensor == [[1, 0, 1]] # If "name" bytes_list is ["bob", "wanda"]
dense_tensor == [[2, 0, 0]] # If "name" bytes_list is ["bob", "bob"]
| Аргументы | |
|---|---|
categorical_column | A CategoricalColumn созданный с помощью функций categorical_column_with_* или crossed_column |
| Возвращаемое значение | |
|---|---|
An IndicatorColumn. |
| Исключения | |
|---|---|
ValueError | Если categorical_column не является типом CategoricalColumn. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/feature_column/indicator_column