Home
last modified time | relevance | path

Searched refs:multi_class_head (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead_test.py761 head = head_lib.multi_class_head(n_classes=n_classes)
776 head = head_lib.multi_class_head(n_classes=len(self._labels) + 1)
785 head = head_lib.multi_class_head(n_classes=n_classes)
814 head = head_lib.multi_class_head(n_classes=2)
824 head = head_lib.multi_class_head(n_classes=n_classes)
841 head = head_lib.multi_class_head(n_classes=n_classes, head_name="head_name")
868 head = head_lib.multi_class_head(n_classes=n_classes,
884 head = head_lib.multi_class_head(n_classes=n_classes)
901 head = head_lib.multi_class_head(n_classes=2, label_name=label_name)
920 head = head_lib.multi_class_head(
[all …]
Dcomposable_model_test.py134 head = head_lib.multi_class_head(n_classes=2)
160 head = head_lib.multi_class_head(n_classes=2)
174 head = head_lib.multi_class_head(n_classes=3)
Ddebug.py196 head_lib.multi_class_head(
D__init__.py318 from tensorflow.contrib.learn.python.learn.estimators.head import multi_class_head
Ddnn_test.py63 'head': head_lib.multi_class_head(2),
94 'head': head_lib.multi_class_head(2),
138 'head': head_lib.multi_class_head(2),
207 head=head_lib.multi_class_head(2, weight_column_name='w'),
Ddnn.py374 head_lib.multi_class_head(
Dlinear.py465 head = head_lib.multi_class_head(
Ddnn_linear_combined.py672 head = head_lib.multi_class_head(
Ddnn_linear_combined_test.py119 'head': head_lib.multi_class_head(2),
153 'head': head_lib.multi_class_head(2),
Dhead.py270 def multi_class_head(n_classes, function
2138 _multi_class_head = multi_class_head
/external/tensorflow/tensorflow/contrib/learn/
DREADME.md88 `tf.contrib.estimator.multi_class_head` and `tf.contrib.estimator.multi_head`.
92 * `multi_class_head`: If you use `tf.contrib.learn.multi_class_head` with
/external/tensorflow/tensorflow/contrib/kernel_methods/python/
Dkernel_estimators.py304 head=head_lib.multi_class_head(
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py126 return head_lib.multi_class_head(
145 return core_head_lib.multi_class_head(
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator.py383 head=head_lib.multi_class_head(
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator.py116 head = head_lib.multi_class_head(
Ddnn_tree_combined_estimator.py478 head = head_lib.multi_class_head(
/external/tensorflow/
DRELEASE.md105 * Switch `multi_class_head` to Head API for Canned Estimator V2.