diff options
author | 2018-06-12 01:52:52 -0700 | |
---|---|---|
committer | 2018-06-12 01:55:52 -0700 | |
commit | da88bfa02f6fb7071a41ff065ec9a918b1e0b1d6 (patch) | |
tree | 9675d9d9151c069b484ca3645fa7973fffc35a89 /tensorflow/contrib/estimator | |
parent | f9ae897fdcba9d1f7aa4ed8e0514022f8e5e70f3 (diff) |
Fixes documentation of multi_label_head to render accepted labels as markdown list
PiperOrigin-RevId: 200181836
Diffstat (limited to 'tensorflow/contrib/estimator')
-rw-r--r-- | tensorflow/contrib/estimator/python/estimator/head.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/estimator/python/estimator/head.py b/tensorflow/contrib/estimator/python/estimator/head.py index b798769d2c..9594e5132f 100644 --- a/tensorflow/contrib/estimator/python/estimator/head.py +++ b/tensorflow/contrib/estimator/python/estimator/head.py @@ -529,6 +529,7 @@ def multi_label_head(n_classes, applications, the shape is `[batch_size, n_classes]`. Labels can be: + * A multi-hot tensor of shape `[D0, D1, ... DN, n_classes]` * An integer `SparseTensor` of class indices. The `dense_shape` must be `[D0, D1, ... DN, ?]` and the values within `[0, n_classes)`. |