Searched refs:multi_label_head (Results 1 – 4 of 4) sorted by relevance
382 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 …]
320 from tensorflow.contrib.learn.python.learn.estimators.head import multi_label_head
388 def multi_label_head(n_classes, function2140 _multi_label_head = multi_label_head
101 * `loss_fn`: Supported for `multi_label_head`. If you need it for other heads,