Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.h478 right_counts_.resize(num_splits(), stats); in ClassificationAddSplitStats()
483 right_counts_.erase(right_counts_.begin() + split_num, in ClassificationRemoveSplitStats()
484 right_counts_.begin() + (split_num + 1)); in ClassificationRemoveSplitStats()
488 right_counts_.clear(); in ClearInternal()
499 right_counts_[split].accumulate(int_label, weight); in ClassificationAddRightExample()
514 return right_counts_[split].get_weight(class_num); in right_count()
519 std::vector<FixedSizeClassStats> right_counts_; variable
Dgrow_stats.cc735 right_counts_.clear(); in ExtractFromProto()
744 right_counts_.emplace_back(params_.num_classes_to_track(), in ExtractFromProto()
746 right_counts_[split_num].ExtractFromProto(right_stats); in ExtractFromProto()
766 right_counts_[split_num].PackToProto(right_stats); in PackToProto()
774 right_counts_[split].set_sum_and_square(right_sum, &right_square); in GiniScore()
791 right_counts_[best_split_index].PackToProto(right_counts); in InitLeafClassStats()