Spec-Zone.ru › TensorFlow 1.15

tf.contrib.layers.weighted_sum_from_feature_columns

Функция для построения линейной модели предсказаний на основе столбцов признаков.

tf.contrib.layers.weighted_sum_from_feature_columns(
    columns_to_tensors, feature_columns, num_outputs, weight_collections=None,
    trainable=True, scope=None
)

Обычно один пример в обучающих данных описывается столбцами признаков. Данная функция генерирует взвешенную сумму для каждого значения num_outputs. Взвешенная сумма соответствует значениям logits в задачах классификации и значениям предсказаний в задачах линейной регрессии.

Пример:

# Building model for training
feature_columns = (
    real_valued_column("my_feature1"),
    ...
)
columns_to_tensor = tf.io.parse_example(...)
logits = weighted_sum_from_feature_columns(
    columns_to_tensors=columns_to_tensor,
    feature_columns=feature_columns,
    num_outputs=1)
loss = tf.nn.sigmoid_cross_entropy_with_logits(labels=labels,
                                               logits=logits)
Аргументы
columns_to_tensors Сопоставление столбца признака и тензора. Ключ 'string' означает базовый признак (не преобразованный). В качестве ключа также может быть использован FeatureColumn. Это означает, что FeatureColumn уже преобразован этапом обработки данных. Например, inflow может выполнить преобразования.
feature_columns Множество, содержащее все столбцы признаков. Все элементы множества должны быть экземплярами классов, производными от FeatureColumn.
num_outputs Целое число, определяющее количество выходных значений. Значение по умолчанию равно 1.
weight_collections Список графовых коллекций, в которые добавляются веса.
trainable Если True , также добавить переменные в графовую коллекцию GraphKeys.TRAINABLE_VARIABLES (см. tf.Variable).
scope Необязательный scope для variable_scope.
Возвращаемые значения
Кортеж, содержащий:
  • Тензор, представляющий предсказания линейной модели.
  • Словарь, сопоставляющий столбец признака соответствующей переменной.
  • Переменная, используемая для смещения (bias).
Исключения
ValueError Если столбец признаков не может быть использован для линейных предсказаний.

© 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/weighted_sum_from_feature_columns

Spec-Zone.ru

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