Spec-Zone.ru › TensorFlow 2.3

tf.feature_column.categorical_column_with_vocabulary_file

Просмотреть исходный код на GitHub

Столбец CategoricalColumn с файлом словаря.

tf.feature_column.categorical_column_with_vocabulary_file(
    key, vocabulary_file, vocabulary_size=None, dtype=tf.dtypes.string,
    default_value=None, num_oov_buckets=0
)

Используйте этот столбец, когда ваши входные данные представлены в строчном или целочисленном формате, и у вас есть файл словаря, который сопоставляет каждое значение целочисленному идентификатору. По умолчанию значения, отсутствующие в словаре, игнорируются. Используйте либо (но не оба) num_oov_buckets и default_value, чтобы указать, как обрабатывать значения, отсутствующие в словаре.

Для входного словаря features, features[key] может быть либо Tensor, либо SparseTensor. Если Tensor, пропущенные значения могут быть представлены как -1 для целых чисел и '' для строк, которые будут отброшены этим столбцом признаков.

Пример с num_oov_buckets: Файл '/us/states.txt' содержит 50 строк, каждая из которых содержит двухсимвольное сокращение штата США. Все входные значения из этого файла присваивают идентификатор 0-49, соответствующий номеру строки. Все остальные значения хэшируются и присваивают идентификатор 50-54.

states = categorical_column_with_vocabulary_file(
    key='states', vocabulary_file='/us/states.txt', vocabulary_size=50,
    num_oov_buckets=5)
columns = [states, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
linear_prediction = linear_model(features, columns)

Пример с default_value: Файл '/us/states.txt' содержит 51 строку - первая строка - 'XX', а остальные 50 содержат двухсимвольное сокращение штата США. И литеральное значение 'XX' во входных данных, и другие отсутствующие в файле значения, будут присвоены идентификатор 0. Все остальные значения получат соответствующий номер строки 1-50.

states = categorical_column_with_vocabulary_file(
    key='states', vocabulary_file='/us/states.txt', vocabulary_size=51,
    default_value=0)
columns = [states, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
linear_prediction, _, _ = linear_model(features, columns)

И для создания встраивания, используя:

columns = [embedding_column(states, 3),...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
dense_tensor = input_layer(features, columns)
Аргументы
key Уникальная строка, идентифицирующая входной признак. Она используется в качестве имени столбца и ключа словаря для конфигурации анализа признаков, объектов признаков Tensor и столбцов признаков.
vocabulary_file Имя файла словаря.
vocabulary_size Количество элементов в словаре. Это значение должно быть не больше длины vocabulary_file, если меньше, то последующие значения игнорируются. Если None, то оно устанавливается в длину vocabulary_file.
dtype Тип признаков. Поддерживаются только строковые и целочисленные типы.
default_value Целочисленное значение идентификатора, которое нужно вернуть для значений признаков, отсутствующих в словаре, по умолчанию -1. Это значение нельзя указать вместе с положительным значением num_oov_buckets.
num_oov_buckets Неотрицательное целое число, количество ведер для значений, отсутствующих в словаре. Все входные значения, отсутствующие в словаре, будут назначены идентификаторам в диапазоне [vocabulary_size, vocabulary_size+num_oov_buckets) на основе хэша входного значения. Положительное значение num_oov_buckets не может быть указано вместе с default_value.
Возвращаемые значения
Столбец CategoricalColumn с файлом словаря.
Исключения
ValueError Файл vocabulary_file отсутствует или не может быть открыт.
ValueError Файл vocabulary_size отсутствует или меньше 1.
ValueError num_oov_buckets - отрицательное целое число.
ValueError num_oov_buckets и default_value указаны одновременно.
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_vocabulary_file

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API