Home
last modified time | relevance | path

Searched refs:total_counts_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.h306 total_counts_.resize(num_outputs_); in Initialize()
324 total_counts_.clear(); in ClearInternal()
336 num_outputs_seen_ += total_counts_[int_label] == 0 && weight > 0; in ClassificationAddTotalExample()
337 total_counts_[int_label] += weight; in ClassificationAddTotalExample()
346 return total_counts_[class_num] - in right_count()
355 std::vector<float> total_counts_; variable
389 total_counts_.clear(); in ClearInternal()
393 bool is_pure() const override { return total_counts_.size() <= 1; } in is_pure()
400 total_counts_[int_label] += weight; in ClassificationAddTotalExample()
409 return total_counts_.at(class_num) - left_counts_[split].at(class_num); in right_count()
[all …]
Dgrow_stats.cc427 total_counts_[i] = class_stats.value(i).float_value(); in ExtractFromProto()
428 num_outputs_seen_ += total_counts_[i] != 0; in ExtractFromProto()
453 class_stats->add_value()->set_float_value(total_counts_[i]); in PackToProto()
500 right_counts->add_value()->set_float_value(total_counts_[i] - in InitLeafClassStats()
517 total_counts_[entry.first] = entry.second.float_value(); in ExtractFromProto()
542 for (const auto& entry : total_counts_) { in PackToProto()
568 for (const auto& entry : total_counts_) { in GiniScore()
601 for (const auto& entry : total_counts_) { in InitLeafClassStats()