Searched refs:node_metadata (Results 1 – 5 of 5) sorted by relevance
155 TreeNodeMetadata* node_metadata = tree_node.mutable_node_metadata(); in RecursivePruneTree() local156 if (node_metadata->gain() < 0 && in RecursivePruneTree()165 (*tree_node.mutable_leaf()) = *node_metadata->mutable_original_leaf(); in RecursivePruneTree()171 node_metadata->clear_original_leaf(); in RecursivePruneTree()751 TreeNodeMetadata* node_metadata = in SplitTreeNode() local753 node_metadata->set_gain(split->gain); in SplitTreeNode()756 (*node_metadata->mutable_original_leaf()) = in SplitTreeNode()881 tree_config->nodes(0).node_metadata().gain() <= 0) { in PruneTree()
463 const auto& node_metadata = node->metadata(); in RecursivelyDoPostPrunePreparation() local464 if (node_metadata.gain() < 0 && IsTerminalSplitNode(tree_id, node_id)) { in RecursivelyDoPostPrunePreparation()473 *node->mutable_leaf() = node_metadata.original_leaf(); in RecursivelyDoPostPrunePreparation()
253 assert tree_node.node_metadata.gain == 0260 tree_node.node_metadata.gain * dtec.tree_weights[tree_idx])
21 TreeNodeMetadata node_metadata = 777; field
782 string node_metadata = GetInstructionNodeMetadata(instr); in DumpInstruction() local810 InstructionId(instr), node_body, node_shape, node_metadata, in DumpInstruction()