Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc221 const int32 best_feature_id = feature_ids(best_candidate.feature_idx); in FindBestSplitsPerNode() local
225 if (feature_id < best_feature_id) { in FindBestSplitsPerNode()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dsplit_handler_ops.cc908 int64 best_feature_id = 0; in ComputeObliviousDecisionTree() local
966 best_feature_id = current_feature_id; in ComputeObliviousDecisionTree()
981 equality_split->set_feature_id(best_feature_id); in ComputeObliviousDecisionTree()