tf.contrib.layers.input_from_feature_columns
Построитель входного слоя стиля tf.contrib.layers, основанный на FeatureColumns.
tf.contrib.layers.input_from_feature_columns(
columns_to_tensors, feature_columns, weight_collections=None, trainable=True,
scope=None, cols_to_outs=None
)
В общем случае, один пример в обучающих данных описывается с помощью столбцов признаков. На первом слое модели эти данные, ориентированные на столбцы, должны быть преобразованы в один тензор. Каждый столбец признаков требует разного вида операции во время этого преобразования. Например, для разреженных признаков требуется совершенно другой подход, чем для непрерывных признаков.
Пример:
# Building model for training
columns_to_tensor = tf.io.parse_example(...)
first_layer = input_from_feature_columns(
columns_to_tensors=columns_to_tensor,
feature_columns=feature_columns)
second_layer = fully_connected(inputs=first_layer, ...)
...
где столбцы признаков могут быть определены следующим образом:
sparse_feature = sparse_column_with_hash_bucket(
column_name="sparse_col", ...)
sparse_feature_emb = embedding_column(sparse_id_column=sparse_feature, ...)
real_valued_feature = real_valued_column(...)
real_valued_buckets = bucketized_column(
source_column=real_valued_feature, ...)
feature_columns=[sparse_feature_emb, real_valued_buckets]
| Аргументы | |
|---|---|
columns_to_tensors | Сопоставление столбца признаков с тензорами. Ключ 'string' означает базовый признак (не преобразованный). Также в качестве ключа может использоваться FeatureColumn. Это означает, что FeatureColumn уже преобразован этапом обработки входных данных. |
feature_columns | Множество, содержащее все столбцы признаков. Все элементы множества должны быть экземплярами классов, производных от FeatureColumn. |
weight_collections | Список коллекций графа, в которые добавляются веса. |
trainable | Если True также добавить переменные в коллекцию графа GraphKeys.TRAINABLE_VARIABLES (см. tf.Variable). |
scope | Необязательный scope для variable_scope. |
cols_to_outs | Необязательный словарь, сопоставляющий столбец признаков с выходным тензором, который конкатенируется в возвращаемый тензор. |
| Возвращаемое значение | |
|---|---|
| Тензор, который можно использовать в скрытых слоях нейронной сети. |
| Исключения | |
|---|---|
ValueError | если FeatureColumn нельзя использовать в нейронной сети. |
© 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/r1.15/api_docs/python/tf/contrib/layers/input_from_feature_columns