OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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