Searched refs:int_label (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | leaf_model_operators.cc | 56 const int32 int_label = target->GetTargetAsClassIndex(example, 0); in UpdateModel() local
57 QCHECK_LT(int_label, params_.num_outputs()) in UpdateModel()
60 QCHECK_GE(int_label, 0); in UpdateModel()
61 auto* val = leaf->mutable_vector()->mutable_value(int_label); in UpdateModel()
91 const int32 int_label = target->GetTargetAsClassIndex(example, 0); in UpdateModel() local
92 QCHECK_LT(int_label, params_.num_outputs()) in UpdateModel()
95 QCHECK_GE(int_label, 0); in UpdateModel()
99 auto it = value_map->find(int_label); in UpdateModel()
101 (*value_map)[int_label].set_float_value(weight); in UpdateModel()
|
| D | grow_stats.h | 203 virtual void ClassificationAddLeftExample(int split, int64 int_label,
205 virtual void ClassificationAddRightExample(int split, int64 int_label, in ClassificationAddRightExample() argument
209 virtual void ClassificationAddTotalExample(int64 int_label, float weight) = 0;
331 void ClassificationAddLeftExample(int split, int64 int_label, in ClassificationAddLeftExample() argument
333 mutable_left_count(split, int_label) += weight; in ClassificationAddLeftExample()
335 void ClassificationAddTotalExample(int64 int_label, float weight) override { in ClassificationAddTotalExample() argument
336 num_outputs_seen_ += total_counts_[int_label] == 0 && weight > 0; in ClassificationAddTotalExample()
337 total_counts_[int_label] += weight; in ClassificationAddTotalExample()
395 void ClassificationAddLeftExample(int split, int64 int_label, in ClassificationAddLeftExample() argument
397 left_counts_[split][int_label] += weight; in ClassificationAddLeftExample()
[all …]
|
| D | grow_stats.cc | 185 const int64 int_label = target->GetTargetAsClassIndex(example, 0); in AddExample() local
192 left_gini_->update(i, left_count(i, int_label), weight); in AddExample()
194 ClassificationAddLeftExample(i, int_label, weight); in AddExample()
197 right_gini_->update(i, right_count(i, int_label), weight); in AddExample()
199 ClassificationAddRightExample(i, int_label, weight); in AddExample()
203 ClassificationAddTotalExample(int_label, weight); in AddExample()
|