Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.cc33 : weight_sum_(0), in GrowStats()
161 bool basic = (weight_sum_ >= split_after_samples_) && !is_pure(); in IsFinished()
205 weight_sum_ += weight; in AddExample()
213 weight_sum_ < prune_sample_epoch_ * prune_check_every_) { in CheckPrune()
274 const float gini_diff_range = weight_sum_ * (1.0 - 1.0 / num_classes); in CheckPruneHoeffding()
275 float epsilon = gini_diff_range * sqrt(half_ln_dominate_frac_ / weight_sum_); in CheckPruneHoeffding()
284 if (weight_sum_ < min_split_samples_ || in CheckFinishEarly()
285 weight_sum_ < finish_sample_epoch_ * finish_check_every_) { in CheckFinishEarly()
299 float range = 0.25 * static_cast<float>(params_.num_outputs()) * weight_sum_; in CheckFinishEarlyHoeffding()
302 range * sqrt(log(1.0 / (1.0 - dominate_fraction_)) / (2.0 * weight_sum_)); in CheckFinishEarlyHoeffding()
[all …]
Dgrow_stats.h58 weight_sum_ = 0; in Clear()
78 float weight_sum() const { return weight_sum_; } in weight_sum()
81 return weight_sum_ > 0 || splits_.size() == num_splits_to_consider_; in IsInitialized()
100 float weight_sum_; variable
126 weight_sum_ += target->GetTargetWeight(example); in AddExample()
132 return weight_sum_ >= split_after_samples_; in IsFinished()
182 return weight_sum_ > 0 || (splits_.size() == num_splits_to_consider_ && in IsInitialized()