def _get_class_id(self, predictions_dict):
    if ((self._learner_config.multi_class_strategy == learner_pb2.LearnerConfig.TREE_PER_CLASS) and (self._logits_dimension != 1)):
        return math_ops.to_int32((predictions_dict[NUM_TREES_ATTEMPTED] % self._logits_dimension))
    return constant_op.constant((- 1), dtype=dtypes.int32)
def _get_class_id(self, predictions_dict):
    if ((self._learner_config.multi_class_strategy == learner_pb2.LearnerConfig.TREE_PER_CLASS) and (self._logits_dimension != 1)):
        return math_ops.to_int32((predictions_dict[NUM_TREES_ATTEMPTED] % self._logits_dimension))
    return constant_op.constant((- 1), dtype=dtypes.int32)