| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| debug.py | 161 label_keys=None):
179 label_keys: Optional list of strings with size `[n_classes]` defining the
192 label_keys=label_keys)}
|
| dnn.py | 255 If the user specifies `label_keys` in constructor, labels must be strings from
256 the `label_keys` vocabulary. Example:
259 label_keys = ['label0', 'label1', 'label2']
263 label_keys=label_keys)
265 def input_fn_train: # returns x, y (where y is one of label_keys).
269 def input_fn_eval: # returns x, y (where y is one of label_keys).
273 # predict_classes returns one of label_keys.
307 label_keys=None):
349 label_keys: Optional list of strings with size `[n_classes]` defining th
[all...] |
| linear.py | 353 If the user specifies `label_keys` in constructor, labels must be strings from
354 the `label_keys` vocabulary. Example:
357 label_keys = ['label0', 'label1', 'label2']
361 label_keys=label_keys)
363 def input_fn_train: # returns x, y (where y is one of label_keys).
367 def input_fn_eval: # returns x, y (where y is one of label_keys).
371 # predict_classes returns one of label_keys.
401 label_keys=None):
436 label_keys: Optional list of strings with size `[n_classes]` defining th
[all...] |
| head.py | 265 label_keys=None):
271 if `label_keys` is specified, then labels must be strings from this
295 label_keys: Optional list of strings with size `[n_classes]` defining the
304 ValueError: If `len(label_keys) != n_classes`.
316 if label_keys:
317 raise ValueError("label_keys is not supported for n_classes=2.")
335 label_keys=label_keys)
[all...] |
| head_test.py | [all...] |
| dnn_linear_combined_test.py | 544 """Tests n_classes > 2 with label_keys vocabulary for labels."""
546 label_keys = [b'label0', b'label1', b'label2']
562 [[label_keys[1]], [label_keys[0]], [label_keys[0]]],
578 label_keys=label_keys)
591 self.assertIn(pred, label_keys)
[all...] |
| dnn_test.py | 559 """Tests n_classes > 2 with label_keys vocabulary for labels."""
561 label_keys = [b'label0', b'label1', b'label2']
577 [[label_keys[1]], [label_keys[0]], [label_keys[0]]],
593 label_keys=label_keys,
607 self.assertIn(pred, label_keys)
[all...] |
| linear_test.py | 177 """Tests n_classes > 2 with label_keys vocabulary for labels."""
179 label_keys = [b'label0', b'label1', b'label2']
191 [[label_keys[1]], [label_keys[0]], [label_keys[0]]],
201 label_keys=label_keys)
214 self.assertIn(pred, label_keys)
[all...] |
| dnn_linear_combined.py | 534 If the user specifies `label_keys` in constructor, labels must be strings from
535 the `label_keys` vocabulary. Example:
538 label_keys = ['label0', 'label1', 'label2']
544 label_keys=label_keys)
546 def input_fn_train: # returns x, y (where y is one of label_keys).
550 def input_fn_eval: # returns x, y (where y is one of label_keys).
554 # predict_classes returns one of label_keys.
595 label_keys=None,
647 label_keys: Optional list of strings with size `[n_classes]` defining th
[all...] |
| debug_test.py | 376 n_classes=3, label_keys=['foo', 'bar', 'baz'])
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| dnn_tree_combined_estimator.py | 271 label_keys=None,
301 label_keys: Optional list of strings with size `[n_classes]` defining the
329 label_keys=label_keys,
|
| estimator.py | 40 label_keys=None,
58 label_keys: Optional list of strings with size `[n_classes]` defining the
|
