def _get_measures(self, targets, predictions, logits): with tf.variable_scope('measures_{}'.format(self.name)): accuracy_val, correct_predictions = get_accuracy( targets, predictions, self.name) hits_at_k_val, mean_hits_at_k = get_hits_at_k( targets, logits, self.top_k, self.name) return correct_predictions, accuracy_val, hits_at_k_val, mean_hits_at_k
def _get_measures(self, targets, predictions): with tf.variable_scope('measures_{}'.format(self.name)): accuracy, correct_predictions = get_accuracy( targets, predictions, self.name ) return correct_predictions, accuracy