Searched refs:dense_split (Results 1 – 4 of 4) sorted by relevance
79 auto* dense_split = tree2->add_nodes()->mutable_dense_float_binary_split(); in TEST_F() local80 dense_split->set_feature_column(0); in TEST_F()81 dense_split->set_threshold(5.0f); in TEST_F()82 dense_split->set_left_id(1); in TEST_F()83 dense_split->set_right_id(2); in TEST_F()173 auto* dense_split = tree2->add_nodes()->mutable_dense_float_binary_split(); in TEST_F() local174 dense_split->set_feature_column(0); in TEST_F()175 dense_split->set_threshold(5.0f); in TEST_F()176 dense_split->set_left_id(1); in TEST_F()177 dense_split->set_right_id(2); in TEST_F()[all …]
285 auto* dense_split = in ComputeNormalDecisionTree() local287 dense_split->set_feature_column(state->feature_column_group_id()); in ComputeNormalDecisionTree()288 dense_split->set_threshold( in ComputeNormalDecisionTree()654 boosted_trees::trees::DenseFloatBinarySplit* dense_split = nullptr; in Compute() local656 dense_split = split_info.mutable_split_node() in Compute()660 dense_split = split_info.mutable_split_node() in Compute()664 dense_split->set_feature_column(state.feature_column_group_id()); in Compute()670 dense_split->set_dimension_id(best_dimension_id); in Compute()671 dense_split->set_threshold(bucket_boundaries(best_bucket_id)); in Compute()
51 const auto& split = current.dense_split(); in TraverseTree()
15 DenseSplit dense_split = 4; field