Home
last modified time | relevance | path

Searched refs:head_lib (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead_test.py28 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
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 …]
Dcomposable_model_test.py26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
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)
Ddebug.py56 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
196 head_lib.multi_class_head(
311 head_lib.regression_head(
Dlinear.py36 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
233 if isinstance(head, head_lib._BinarySvmHead): # pylint: disable=protected-access
235 elif isinstance(head, head_lib._BinaryLogisticHead): # pylint: disable=protected-access
237 elif isinstance(head, head_lib._RegressionHead): # pylint: disable=protected-access
465 head = head_lib.multi_class_head(
737 head = head_lib.regression_head(
Dsvm.py30 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
151 "head": head_lib.binary_svm_head(
Ddnn.py36 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
374 head_lib.multi_class_head(
657 head_lib.regression_head(
Ddnn_linear_combined.py37 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
672 head = head_lib.multi_class_head(
1008 head = head_lib.regression_head(
Ddnn_test.py35 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
63 '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'),
Dlinear_test.py33 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
1883 head=head_lib.regression_head()),
1909 head=head_lib.regression_head())
1936 head=head_lib.poisson_regression_head())
1954 head=head_lib.regression_head(label_dimension=1),
Ddnn_linear_combined_test.py33 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
67 class _CheckCallsHead(head_lib.Head):
119 'head': head_lib.multi_class_head(2),
153 'head': head_lib.multi_class_head(2),
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/
Dsdca_estimator.py24 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
38 return isinstance(head, head_lib._BinaryLogisticHead) or isinstance(
39 head, head_lib._BinarySvmHead) or isinstance(head,
40 head_lib._RegressionHead)
122 if isinstance(head, head_lib._BinarySvmHead): # pylint: disable=protected-access
124 elif isinstance(head, head_lib._BinaryLogisticHead): # pylint: disable=protected-access
126 elif isinstance(head, head_lib._RegressionHead): # pylint: disable=protected-access
383 head=head_lib.multi_class_head(
515 head=head_lib.regression_head(weight_column_name=weight_column_name),
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Ddistillation_loss.py36 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
52 return head_lib._mean_squared_loss( # pylint: disable=protected-access
65 return head_lib._log_loss_with_two_classes( # pylint: disable=protected-access
70 return head_lib._softmax_cross_entropy_loss( # pylint: disable=protected-access
Dcustom_loss_head.py21 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
27 class CustomLossHead(head_lib._RegressionHead): # pylint: disable=protected-access
Ddnn_tree_combined_estimator_test.py29 from tensorflow.python.estimator.canned import head as head_lib unknown
143 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
174 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
203 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
Ddnn_tree_combined_estimator.py36 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
278 weight_tensor = head_lib._weight_tensor(features, head._weight_column)
280 weight_tensor = head_lib._weight_tensor(
478 head = head_lib.multi_class_head(
604 head = head_lib.regression_head(
Destimator_test.py27 from tensorflow.python.estimator.canned import head as head_lib unknown
193 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
257 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
478 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
508 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
611 head_fn = head_lib._binary_logistic_head_with_sigmoid_cross_entropy_loss(
Destimator.py26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
116 head = head_lib.multi_class_head(
211 head = head_lib.regression_head(
/external/tensorflow/tensorflow/contrib/kernel_methods/python/
Dkernel_estimators.py26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
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/tensor_forest/client/
Drandom_forest_test.py26 from tensorflow.python.estimator.canned import head as head_lib unknown
171 head_fn = head_lib._multi_class_head_with_softmax_cross_entropy_loss(
198 head_fn = head_lib._regression_head(
224 head_fn = head_lib._multi_class_head_with_softmax_cross_entropy_loss(
333 head_fn = head_lib._multi_class_head_with_softmax_cross_entropy_loss(
Drandom_forest.py24 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
120 return head_lib.regression_head(
126 return head_lib.multi_class_head(
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
Destimator_test.py26 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib unknown
89 train_op_fn=head_lib.no_op_train_fn,
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead.py24 from tensorflow.python.estimator.canned import head as head_lib unknown
51 class TimeSeriesRegressionHead(head_lib._Head): # pylint:disable=protected-access
86 head_lib._summary_key(self._name, metric_keys.MetricKeys.LOSS),