HomeSort by relevance Sort by last modified time
    Searched refs:CLASS_IDS (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/python/estimator/canned/
prediction_keys.py 30 CLASS_IDS = 'class_ids'
head.py 626 self, labels, class_ids, weights, unreduced_loss, regularization_loss):
630 (labels, class_ids, weights, unreduced_loss, regularization_loss)):
643 predictions=class_ids,
728 # class_ids's shape is [D0, D1, ... DN].
729 class_ids = math_ops.argmax(logits, axis=-1, name=pred_keys.CLASS_IDS)
730 class_ids = array_ops.expand_dims(class_ids, axis=-1)
735 classes = table.lookup(class_ids)
737 classes = string_ops.as_string(class_ids, name='str_classes'
    [all...]
dnn_testing_utils.py     [all...]
head_test.py 364 predictions[prediction_keys.PredictionKeys.CLASS_IDS])
    [all...]

Completed in 479 milliseconds