Skip to content

Commit da88bfa

Browse files
Fixes documentation of multi_label_head to render accepted labels as markdown list
PiperOrigin-RevId: 200181836
1 parent f9ae897 commit da88bfa

File tree

1 file changed

+1
-0
lines changed
  • tensorflow/contrib/estimator/python/estimator

1 file changed

+1
-0
lines changed

tensorflow/contrib/estimator/python/estimator/head.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,7 @@ def multi_label_head(n_classes,
529529
applications, the shape is `[batch_size, n_classes]`.
530530
531531
Labels can be:
532+
532533
* A multi-hot tensor of shape `[D0, D1, ... DN, n_classes]`
533534
* An integer `SparseTensor` of class indices. The `dense_shape` must be
534535
`[D0, D1, ... DN, ?]` and the values within `[0, n_classes)`.

0 commit comments

Comments
 (0)