Home
last modified time | relevance | path

Searched refs:multi_label_head (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead_test.py382 head = head_lib.multi_label_head(
399 head = head_lib.multi_label_head(
426 head = head_lib.multi_label_head(n_classes=len(self._labels[0]) + 1)
435 head = head_lib.multi_label_head(
472 head = head_lib.multi_label_head(
483 head = head_lib.multi_label_head(
499 head = head_lib.multi_label_head(
538 head = head_lib.multi_label_head(n_classes=n_classes, head_name="head_name")
569 head = head_lib.multi_label_head(
586 head = head_lib.multi_label_head(
[all …]
D__init__.py320 from tensorflow.contrib.learn.python.learn.estimators.head import multi_label_head
Dhead.py388 def multi_label_head(n_classes, function
2140 _multi_label_head = multi_label_head
/external/tensorflow/tensorflow/contrib/learn/
DREADME.md101 * `loss_fn`: Supported for `multi_label_head`. If you need it for other heads,