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

  /external/tensorflow/tensorflow/python/estimator/canned/
head.py 726 pred_keys = prediction_keys.PredictionKeys
729 class_ids = math_ops.argmax(logits, axis=-1, name=pred_keys.CLASS_IDS)
739 probabilities = nn.softmax(logits, name=pred_keys.PROBABILITIES)
741 pred_keys.LOGITS: logits,
742 pred_keys.PROBABILITIES: probabilities,
744 pred_keys.CLASS_IDS: class_ids,
745 pred_keys.CLASSES: classes,
    [all...]
  /external/tensorflow/tensorflow/contrib/estimator/python/estimator/
head.py 451 pred_keys = prediction_keys.PredictionKeys
453 probabilities = math_ops.sigmoid(logits, name=pred_keys.PROBABILITIES)
455 pred_keys.LOGITS: logits,
456 pred_keys.PROBABILITIES: probabilities,

Completed in 98 milliseconds