Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc601 void GetParentWeightedMean(float leaf_sum, const float* leaf_data, in GetParentWeightedMean() argument
606 if (leaf_sum < valid_leaf_threshold && parent_sum >= 0) { in GetParentWeightedMean()
608 << "child sum = " << leaf_sum; in GetParentWeightedMean()
613 std::min(1.0f, (valid_leaf_threshold - leaf_sum) / parent_sum); in GetParentWeightedMean()
614 leaf_sum += parent_weight * parent_sum; in GetParentWeightedMean()
615 VLOG(1) << "Sum w/ parent included = " << leaf_sum; in GetParentWeightedMean()
623 (*mean)[c] = w / leaf_sum; in GetParentWeightedMean()
Dtree_utils.h297 void GetParentWeightedMean(float leaf_sum, const float* leaf_data,