Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc96 SplitCandidate& best_split = best_split_it->second; in UpdateBestSplit() local
97 if (TF_PREDICT_FALSE(split->gain == best_split.gain)) { in UpdateBestSplit()
102 << best_split.split_info.split_node().node_case() << ")"; in UpdateBestSplit()
104 best_split.split_info.split_node().node_case()) { in UpdateBestSplit()
105 best_split = std::move(*split); in UpdateBestSplit()
107 best_split.split_info.split_node().node_case()) { in UpdateBestSplit()
111 << ", best split: " << best_split.handler_id << ")"; in UpdateBestSplit()
112 if (split->handler_id > best_split.handler_id) { in UpdateBestSplit()
113 best_split = std::move(*split); in UpdateBestSplit()
116 } else if (split->gain > best_split.gain) { in UpdateBestSplit()
[all …]