blob: 850340f5c2096ca674616254de45d96b84200a64 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
syntax = "proto3";
option cc_enable_arenas = true;
package tensorflow.boosted_trees.learner;
import "tensorflow/contrib/boosted_trees/proto/tree_config.proto";
// Gathered information for a split node.
message SplitInfo {
// The split node without the child nodes attached.
tensorflow.boosted_trees.trees.TreeNode split_node = 1;
// Left Leaf node.
tensorflow.boosted_trees.trees.Leaf left_child = 2;
// Right Leaf node.
tensorflow.boosted_trees.trees.Leaf right_child = 3;
}
message ObliviousSplitInfo {
// The split node with the feature_column and threshold defined.
tensorflow.boosted_trees.trees.TreeNode split_node = 1;
// The new leaves of the tree.
repeated tensorflow.boosted_trees.trees.Leaf children_leaves = 2;
}
|