Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.h317 left_counts_.resize(num_outputs_ * num_splits()); in ClassificationAddSplitStats()
320 left_counts_.erase(left_counts_.begin() + num_outputs_ * split_num, in ClassificationRemoveSplitStats()
321 left_counts_.begin() + num_outputs_ * (split_num + 1)); in ClassificationRemoveSplitStats()
325 left_counts_.clear(); in ClearInternal()
343 return left_counts_[split * num_outputs_ + class_num]; in left_count()
347 left_counts_[split * num_outputs_ + class_num]; in right_count()
352 return left_counts_[split * num_outputs_ + class_num]; in mutable_left_count()
363 std::vector<float> left_counts_; variable
382 left_counts_.resize(num_splits()); in ClassificationAddSplitStats()
385 left_counts_.erase(left_counts_.begin() + split_num, in ClassificationRemoveSplitStats()
[all …]
Dgrow_stats.cc527 left_counts_[split_num][entry.first] = val; in ExtractFromProto()
555 for (const auto& entry : left_counts_[split_num]) { in PackToProto()
572 auto it = left_counts_[split].find(label); in GiniScore()
573 if (it == left_counts_[split].end()) { in GiniScore()
602 auto it = left_counts_[best_split_index].find(entry.first); in InitLeafClassStats()
603 if (it == left_counts_[best_split_index].end()) { in InitLeafClassStats()
734 left_counts_.clear(); in ExtractFromProto()
739 left_counts_.emplace_back(params_.num_classes_to_track(), in ExtractFromProto()
741 left_counts_[split_num].ExtractFromProto(left_stats); in ExtractFromProto()
762 left_counts_[split_num].PackToProto(left_stats); in PackToProto()
[all …]