tf.feature_column.categorical_column_with_hash_bucket
| Просмотреть исходный код на GitHub |
Представляет разреженное свойство, где идентификаторы устанавливаются с помощью хэширования.
tf.feature_column.categorical_column_with_hash_bucket(
key, hash_bucket_size, dtype=tf.dtypes.string
)
Используйте этот метод, когда ваши разреженные свойства представлены в строчном или целочисленном формате, и вы хотите распределить свои входные данные в конечное число ведер с помощью хэширования. output_id = Hash(input_feature_string) % bucket_size для входных данных строкового типа. Для входных данных целочисленного типа значение сначала преобразуется в строковое представление, а затем хэшируется по той же формуле.
Для входного словаря features, features[key] является либо Tensor, либо SparseTensor. Если Tensor, пропущенные значения могут быть представлены -1 для целых чисел и '' для строк, которые будут отброшены этим свойством.
Пример:
keywords = categorical_column_with_hash_bucket("keywords", 10K)
columns = [keywords, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
linear_prediction = linear_model(features, columns)
# or
keywords_embedded = embedding_column(keywords, 16)
columns = [keywords_embedded, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
dense_tensor = input_layer(features, columns)
| Аргументы | |
|---|---|
key | Уникальная строка, идентифицирующая входное свойство. Она используется в качестве имени столбца и ключа словаря для конфигурации обработки свойств, объектов свойств Tensor и свойств столбцов. |
hash_bucket_size | Целое число > 1. Количество ведер. |
dtype | Тип свойств. Поддерживаются только строковый и целочисленный типы. |
| Возвращает | |
|---|---|
HashedCategoricalColumn. |
| Исключения | |
|---|---|
ValueError | hash_bucket_size не больше 1. |
ValueError | dtype не является ни строкой, ни целым числом. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/feature_column/categorical_column_with_hash_bucket