HomeSort by relevance Sort by last modified time
    Searched refs:head_lib (Results 1 - 25 of 33) sorted by null

1 2

  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
head_test.py 28 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
116 head = head_lib.poisson_regression_head()
124 train_op_fn=head_lib.no_op_train_fn,
144 head = head_lib.regression_head()
150 train_op_fn=head_lib.no_op_train_fn,
158 head = head_lib.regression_head(link_fn=math_ops.square)
168 train_op_fn=head_lib.no_op_train_fn,
177 head = head_lib.regression_head()
184 train_op_fn=head_lib.no_op_train_fn,
188 head = head_lib.regression_head(
    [all...]
debug.py 52 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
188 head_lib.multi_class_head(
299 head_lib.regression_head(
composable_model_test.py 26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
134 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)
svm.py 25 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
142 "head": head_lib.binary_svm_head(
linear.py 32 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
228 if isinstance(head, head_lib._BinarySvmHead): # pylint: disable=protected-access
230 elif isinstance(head, head_lib._BinaryLogisticHead): # pylint: disable=protected-access
232 elif isinstance(head, head_lib._RegressionHead): # pylint: disable=protected-access
454 head = head_lib.multi_class_head(
722 head = head_lib.regression_head(
    [all...]
  /external/tensorflow/tensorflow/contrib/estimator/python/estimator/
head.py 22 from tensorflow.python.estimator.canned import head as head_lib
98 return head_lib._multi_class_head_with_softmax_cross_entropy_loss( # pylint:disable=protected-access
167 return head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss( # pylint:disable=protected-access
222 return head_lib._regression_head_with_mean_squared_error_loss( # pylint:disable=protected-access
316 head_lib._validate_loss_fn_args(loss_fn) # pylint:disable=protected-access
326 class _MultiLabelHead(head_lib._Head): # pylint:disable=protected-access
389 return head_lib._assert_range(labels, 2, message=err_msg) # pylint:disable=protected-access,
396 processed_labels = head_lib._check_dense_labels_match_logits_and_reshape( # pylint:disable=protected-access
400 unweighted_loss = head_lib._call_loss_fn( # pylint:disable=protected-access
410 weights = head_lib._get_weights_and_check_match_logits( # pylint:disable=protected-access
    [all...]
multi_head_test.py 24 from tensorflow.contrib.estimator.python.estimator import head as head_lib
88 head1 = head_lib.multi_label_head(n_classes=2, name='head1')
89 head2 = head_lib.multi_label_head(n_classes=3)
95 head1 = head_lib.multi_label_head(n_classes=2, name='head1')
96 head2 = head_lib.multi_label_head(n_classes=3, name='head2')
104 head1 = head_lib.multi_label_head(n_classes=2, name='head1')
105 head2 = head_lib.multi_label_head(n_classes=3, name='head2')
111 head1 = head_lib.multi_label_head(n_classes=2, name='head1')
112 head2 = head_lib.multi_label_head(n_classes=3, name='head2')
164 head1 = head_lib.multi_label_head(n_classes=2, name='head1'
    [all...]
head_test.py 24 from tensorflow.contrib.estimator.python.estimator import head as head_lib
103 head_lib.multi_label_head(n_classes=None)
109 head_lib.multi_label_head(n_classes=1)
115 head_lib.multi_label_head(n_classes=2, thresholds=[0., 0.5])
121 head_lib.multi_label_head(n_classes=2, thresholds=[0.5, 1.0])
128 head_lib.multi_label_head(n_classes=2, label_vocabulary={'foo': 'bar'})
134 head_lib.multi_label_head(n_classes=3, label_vocabulary=['foo', 'bar'])
139 head_lib.multi_label_head(
143 head_lib.multi_label_head(
153 head_lib.multi_label_head(n_classes=3, loss_fn=_loss_fn
    [all...]
dnn_linear_combined_test.py 28 from tensorflow.contrib.estimator.python.estimator import head as head_lib
54 head=head_lib.regression_head(
102 head=head_lib.regression_head(
157 head=head_lib.regression_head(label_dimension=label_dimension),
dnn_test.py 28 from tensorflow.contrib.estimator.python.estimator import head as head_lib
43 head=head_lib.regression_head(
91 head=head_lib.regression_head(label_dimension=label_dimension),
linear_test.py 27 from tensorflow.contrib.estimator.python.estimator import head as head_lib
44 head=head_lib.regression_head(
92 head=head_lib.regression_head(label_dimension=label_dimension),
multi_head.py 24 from tensorflow.python.estimator.canned import head as head_lib
164 class _MultiHead(head_lib._Head): # pylint:disable=protected-access
223 return head_lib.LossSpec(
355 loss_name = head_lib._summary_key( # pylint:disable=protected-access
  /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
custom_loss_head.py 21 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
27 class CustomLossHead(head_lib._RegressionHead): # pylint: disable=protected-access
estimator.py 24 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
80 head = head_lib.multi_class_head(
148 head = head_lib.regression_head(
dnn_tree_combined_estimator.py 35 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
326 head = head_lib.multi_class_head(
415 head = head_lib.regression_head(
  /external/tensorflow/tensorflow/contrib/linear_optimizer/python/
sdca_estimator.py 24 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
38 return isinstance(head, head_lib._BinaryLogisticHead) or isinstance(
39 head, head_lib._BinarySvmHead) or isinstance(head,
40 head_lib._RegressionHead)
121 if isinstance(head, head_lib._BinarySvmHead): # pylint: disable=protected-access
123 elif isinstance(head, head_lib._BinaryLogisticHead): # pylint: disable=protected-access
125 elif isinstance(head, head_lib._RegressionHead): # pylint: disable=protected-access
373 head=head_lib.multi_class_head(
501 head=head_lib.regression_head(weight_column_name=weight_column_name),
  /external/tensorflow/tensorflow/python/estimator/canned/
head_test.py 27 from tensorflow.python.estimator.canned import head as head_lib
96 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
101 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
107 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
111 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
121 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
131 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
137 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
146 head_lib._multi_class_head_with_softmax_cross_entropy_loss(
151 head = head_lib._multi_class_head_with_softmax_cross_entropy_loss(n_classes
    [all...]
baseline.py 51 from tensorflow.python.estimator.canned import head as head_lib
256 head = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss( # pylint: disable=protected-access
261 head = head_lib._multi_class_head_with_softmax_cross_entropy_loss( # pylint: disable=protected-access
347 head = head_lib._regression_head_with_mean_squared_error_loss( # pylint: disable=protected-access
linear.py 26 from tensorflow.python.estimator.canned import head as head_lib
298 head = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss( # pylint: disable=protected-access
303 head = head_lib._multi_class_head_with_softmax_cross_entropy_loss( # pylint: disable=protected-access
425 head = head_lib._regression_head_with_mean_squared_error_loss( # pylint: disable=protected-access
dnn.py 25 from tensorflow.python.estimator.canned import head as head_lib
139 head: A `head_lib._Head` instance.
331 head = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss( # pylint: disable=protected-access
336 head = head_lib._multi_class_head_with_softmax_cross_entropy_loss( # pylint: disable=protected-access
491 head=head_lib. # pylint: disable=protected-access
dnn_linear_combined.py 27 from tensorflow.python.estimator.canned import head as head_lib
377 head = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss( # pylint: disable=protected-access
382 head = head_lib._multi_class_head_with_softmax_cross_entropy_loss( # pylint: disable=protected-access
557 head=head_lib. # pylint: disable=protected-access
  /external/tensorflow/tensorflow/contrib/kernel_methods/python/
kernel_estimators.py 26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
53 return isinstance(head, head_lib._BinaryLogisticHead) or isinstance(
54 head, head_lib._MultiClassHead)
304 head=head_lib.multi_class_head(
  /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
estimator_test.py 26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
89 train_op_fn=head_lib.no_op_train_fn,
  /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
gan_estimator_impl.py 27 from tensorflow.contrib.gan.python.estimator.python import head as head_lib
161 gan_head = head_lib.gan_head(
  /external/tensorflow/tensorflow/contrib/tensor_forest/client/
random_forest.py 23 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
111 return head_lib.regression_head(
117 return head_lib.multi_class_head(

Completed in 456 milliseconds

1 2