tf.feature_column.categorical_column_with_identity

A CategoricalColumn that returns identity values.

tf.feature_column.categorical_column_with_identity(
    key, num_buckets, default_value=None
)

Используйте этот метод, когда ваши входные данные — целые числа в диапазоне [0, num_buckets), и вы хотите использовать само значение входа в качестве категорического идентификатора. Значения вне этого диапазона приведут к default_value если указано, иначе произойдёт ошибка.

Обычно это используется для непрерывных диапазонов целочисленных индексов, но не обязательно. Однако это может быть неэффективно, если многие идентификаторы не используются. В этом случае рассмотрите categorical_column_with_hash_bucket.

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

В следующих примерах каждому входу в диапазоне [0, 1000000) присваивается то же самое значение. Все остальные входы получают значение default_value 0. Обратите внимание, что буквальный 0 в входных данных приведет к тому же идентификатору по умолчанию.

Линейная модель:

video_id = categorical_column_with_identity(
    key='video_id', num_buckets=1000000, default_value=0)
columns = [video_id, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
linear_prediction, _, _ = linear_model(features, columns)

Встраивание для модели DNN:

columns = [embedding_column(video_id, 9),...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
dense_tensor = input_layer(features, columns)