1syntax = "proto3"; 2 3option cc_enable_arenas = true; 4 5package tensorflow.boosted_trees.learner; 6 7import "tensorflow/contrib/boosted_trees/proto/tree_config.proto"; 8 9// Gathered information for a split node. 10message SplitInfo { 11 // The split node without the child nodes attached. 12 tensorflow.boosted_trees.trees.TreeNode split_node = 1; 13 14 // Left Leaf node. 15 tensorflow.boosted_trees.trees.Leaf left_child = 2; 16 17 // Right Leaf node. 18 tensorflow.boosted_trees.trees.Leaf right_child = 3; 19} 20 21message ObliviousSplitInfo { 22 tensorflow.boosted_trees.trees.TreeNode split_node = 1; 23 repeated tensorflow.boosted_trees.trees.Leaf children = 2; 24 // For each child, children_parent_id stores the node_id of its parent when it 25 // was a leaf. For the idx-th child it corresponds the idx/2-th 26 // children_parent_id. 27 repeated int32 children_parent_id = 3; 28} 29