tf.contrib.learn.regression_head
Создаёт Head для линейной регрессии. (устарело)
tf.contrib.learn.regression_head(
label_name=None, weight_column_name=None, label_dimension=1,
enable_centered_bias=False, head_name=None, link_fn=None
)
| Аргументы | |
|---|---|
label_name | Строка, имя ключа в словаре меток. Может быть null, если метка — это тензор (модели с одной головой). |
weight_column_name | Строка, определяющая имя столбца признаков, представляющего веса. Используется для уменьшения или увеличения весов примеров во время обучения. Будет умножаться на потерю примера. |
label_dimension | Количество меток регрессии на пример. Это размер последнего измерения меток Tensor (обычно, эта форма [batch_size, label_dimension]). |
enable_centered_bias | Булево значение. Если True, оценщик обучит переменную смещения, центрированную для каждого класса. Остальная часть структуры модели обучит остаток после центрированного смещения. |
head_name | Имя головы. Если указано, ключи прогнозов, сводок и метрик будут дополнены суффиксом "/" + head_name , а область видимости переменных по умолчанию будет head_name. |
link_fn | Функция связи для преобразования логарифмов в прогнозы. Если предоставлено, эта функция связи будет использоваться вместо тождественной. |
| Возвращаемое значение | |
|---|---|
Экземпляр Head для линейной регрессии. |
© 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/learn/regression_head