tensorflow:: ops:: InTopKV2
#include <nn_ops.h>
Says whether the targets are in the top K
predictions.
Summary
This outputs a batch_size
bool array, an entry out[i]
is true
if the prediction for the target class is among the top k
predictions among all predictions for example i
. Note that the behavior of InTopK
differs from the TopK
op in its handling of ties; if multiple classes have the same prediction value and straddle the top-k
boundary, all of those classes are considered to be in the top k
.
More formally, let
\(predictions_i\) be the predictions for all classes for example i
, \(targets_i\) be the target class for example i
, \(out_i\) be the output for example i
,
$$out_i = predictions_{i, targets_i} \in TopKIncludingTies(predictions_i)$$
Args:
- scope: A Scope object
- predictions: A
batch_size
xclasses
tensor. - targets: A
batch_size
vector of class ids. - k: Number of top elements to look at for computing precision.
Returns:
Constructors and Destructors |
|
---|---|
InTopKV2(const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, ::tensorflow::Input k)
|
Public attributes |
|
---|---|
operation
|
|
precision
|
Public functions |
|
---|---|
node() const
|
::tensorflow::Node *
|
operator::tensorflow::Input() const
|
|
operator::tensorflow::Output() const
|
|
Public attributes
operation
Operation operation
precision
::tensorflow::Output precision
Public functions
InTopKV2
InTopKV2( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, ::tensorflow::Input k )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const