Searched refs:left_stats (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.cc | 435 const auto& left_stats = cand.left_stats().classification().dense_counts(); in ExtractFromProto() local
437 const float val = left_stats.value(i).float_value(); in ExtractFromProto()
459 auto* left_stats = cand->mutable_left_stats() in PackToProto() local
463 left_stats->add_value()->set_float_value(left_count(split_num, i)); in PackToProto()
490 int best_split_index, LeafStat* left_stats, LeafStat* right_stats) const { in InitLeafClassStats() argument
491 auto* left_class_stats = left_stats->mutable_classification(); in InitLeafClassStats()
524 const auto& left_stats = cand.left_stats().classification().sparse_counts(); in ExtractFromProto() local
525 for (auto const& entry : left_stats.sparse_value()) { in ExtractFromProto()
551 auto* left_stats = cand->mutable_left_stats() in PackToProto() local
558 (*left_stats)[entry.first] = val; in PackToProto()
[all …]
|
| D | grow_stats.h | 190 virtual void InitLeafClassStats(int best_split_index, LeafStat* left_stats,
312 void InitLeafClassStats(int best_split_index, LeafStat* left_stats,
377 void InitLeafClassStats(int best_split_index, LeafStat* left_stats,
470 void InitLeafClassStats(int best_split_index, LeafStat* left_stats,
|
| D | decision-tree-resource.cc | 64 model_op_->ExportModel(best->left_stats(), left_leaf); in SplitNode()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/ |
| D | split-stats.h | 47 const NodeStats& left_stats, const NodeStats& right_stats) in SplitStats()
49 left_node_stats(left_stats), in SplitStats()
51 gain(left_stats.gain + right_stats.gain - root_stats.gain - in SplitStats()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | split_handler_ops.cc | 273 NodeStats left_stats = state->ComputeNodeStats(left_gradient_stats); in ComputeNormalDecisionTree() local
277 if (left_stats.gain + right_stats.gain > best_gain) { in ComputeNormalDecisionTree()
278 best_gain = left_stats.gain + right_stats.gain; in ComputeNormalDecisionTree()
279 best_left_node_stats = left_stats; in ComputeNormalDecisionTree()
847 NodeStats left_stats = state->ComputeNodeStats(left_gradient_stats); in ComputeNormalDecisionTree() local
850 left_stats.gain + right_stats.gain > best_gain) { in ComputeNormalDecisionTree()
851 best_gain = left_stats.gain + right_stats.gain; in ComputeNormalDecisionTree()
852 best_left_node_stats = left_stats; in ComputeNormalDecisionTree()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/proto/ |
| D | fertile_stats.proto | 85 LeafStat left_stats = 4; field
|