Searched refs:tree_metadata (Results 1 – 10 of 10) sorted by relevance
230 tree_ensemble_config.tree_metadata.add().is_finalized = True258 tree_ensemble_config.tree_metadata.add().is_finalized = True289 tree_ensemble_config.tree_metadata.add().is_finalized = True294 tree_ensemble_config.tree_metadata.add().is_finalized = True338 tree_ensemble_config.tree_metadata.add().is_finalized = True385 tree_ensemble_config.tree_metadata.add().is_finalized = True390 tree_ensemble_config.tree_metadata.add().is_finalized = True473 tree_ensemble_config.tree_metadata.add().is_finalized = True478 tree_ensemble_config.tree_metadata.add().is_finalized = False520 tree_ensemble_config.tree_metadata.add().is_finalized = True[all …]
222 tree_ensemble_config.tree_metadata.add().is_finalized = True
2452 1, tree_ensemble_config.tree_metadata[0].num_tree_weight_updates)2454 6, tree_ensemble_config.tree_metadata[1].num_tree_weight_updates)2456 2, tree_ensemble_config.tree_metadata[2].num_tree_weight_updates)
626 boosted_trees::trees::DecisionTreeMetadata* const tree_metadata = in UpdateAndRetrieveGrowableTree() local628 tree_metadata->set_is_finalized(max_tree_depth <= 1); in UpdateAndRetrieveGrowableTree()629 tree_metadata->set_num_tree_weight_updates(1); in UpdateAndRetrieveGrowableTree()632 boosted_trees::trees::DecisionTreeMetadata* const tree_metadata = in UpdateAndRetrieveGrowableTree() local634 const auto new_num_layers = tree_metadata->num_layers_grown() + 1; in UpdateAndRetrieveGrowableTree()638 tree_metadata->set_num_layers_grown(new_num_layers); in UpdateAndRetrieveGrowableTree()639 tree_metadata->set_is_finalized(new_num_layers >= max_tree_depth); in UpdateAndRetrieveGrowableTree()960 for (const auto& tree_metadata : ensemble_config.tree_metadata()) { in Compute() local961 num_layers += tree_metadata.num_layers_grown(); in Compute()967 ? ensemble_config.tree_metadata(tree_metadata_size - 1) in Compute()
84 !config.tree_metadata(tree).is_finalized()) { in CalculateTreesToInclude()
129 return tree_ensemble_->tree_metadata(tree_id).num_layers_grown(); in GetNumLayersGrown()210 return tree_ensemble_->tree_metadata(tree_id).is_finalized(); in IsTreeFinalized()216 return tree_ensemble_->tree_metadata(tree_id).post_pruned_nodes_meta_size() > in IsTreePostPruned()406 tree_ensemble_->tree_metadata(tree_id).post_pruned_nodes_meta_size()); in GetPostPruneCorrection()408 tree_ensemble_->tree_metadata(tree_id).post_pruned_nodes_meta( in GetPostPruneCorrection()
151 repeated TreeMetadata tree_metadata = 3; field
151 const int32 num_updates = decision_tree_ensemble_->tree_metadata(index) in SetTreeWeight()
168 repeated DecisionTreeMetadata tree_metadata = 3; field
1713 metadata = tree_ensemble_config.tree_metadata.add()