tf.contrib.distributions.estimator_head_distribution_regression
Создаёт Head для регрессии в рамках обобщённого распределения. (устарело)
tf.contrib.distributions.estimator_head_distribution_regression(
make_distribution_fn, label_dimension=1, logits_dimension=None, label_name=None,
weight_column_name=None, enable_centered_bias=False, head_name=None
)
| Аргументы | |
|---|---|
make_distribution_fn | Python callable который возвращает экземпляр tf.Distribution созданный только с использованием логарифмов. |
label_dimension | Количество меток регрессии на пример. Это размер последнего измерения меток Tensor (как правило, имеет форму [batch_size, label_dimension]). |
logits_dimension | Количество логарифмов на пример. Это размер последнего измерения логарифмов Tensor (как правило, имеет форму [batch_size, logits_dimension]). Значение по умолчанию: label_dimension. |
label_name | Python str, имя ключа в метках dict. Может быть None если метки являются Tensor (модели с единственной головой). |
weight_column_name | Python str определяет имя столбца признаков, представляющее веса. Используется для снижения или увеличения весов примеров во время обучения. Будет умножаться на потерю примера. |
enable_centered_bias | Python bool. Если True, оценщик будет учиться смещенной переменной для каждого класса. Остальная часть структуры модели обучается остатку после центрированной смещенной переменной. |
head_name | Python str, имя головы. Ключи предсказаний, сводок и метрик дополняются префиксом "/" + head_name, а область переменных по умолчанию составляет head_name. |
| Возвращаемое значение | |
|---|---|
Экземпляр 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/distributions/estimator_head_distribution_regression