tf.contrib.distributions.estimator_head_distribution_regression
Creates a Head
for regression under a generic distribution. (deprecated)
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 )
Args | |
---|---|
make_distribution_fn | Python callable which returns a tf.Distribution instance created using only logits. |
label_dimension | Number of regression labels per example. This is the size of the last dimension of the labels Tensor (typically, this has shape [batch_size, label_dimension] ). |
logits_dimension | Number of logits per example. This is the size of the last dimension of the logits Tensor (typically, this has shape [batch_size, logits_dimension] ). Default value: label_dimension . |
label_name | Python str , name of the key in label dict . Can be None if label is a Tensor (single headed models). |
weight_column_name | Python str defining feature column name representing weights. It is used to down weight or boost examples during training. It will be multiplied by the loss of the example. |
enable_centered_bias | Python bool . If True , estimator will learn a centered bias variable for each class. Rest of the model structure learns the residual after centered bias. |
head_name | Python str , name of the head. Predictions, summary and metrics keys are suffixed by "/" + head_name and the default variable scope is head_name . |
Returns | |
---|---|
An instance of Head for generic regression. |
© 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