Searched refs:split_info (Results 1 – 9 of 9) sorted by relevance
83 split_info = split_info_pb2.SplitInfo()84 split_info.ParseFromString(splits[0])85 left_child = split_info.left_child.vector86 right_child = split_info.right_child.vector87 split_node = split_info.split_node.dense_float_binary_split100 split_info = split_info_pb2.SplitInfo()101 split_info.ParseFromString(splits[1])102 left_child = split_info.left_child.vector103 right_child = split_info.right_child.vector104 split_node = split_info.split_node.dense_float_binary_split[all …]
127 split_info = split_info_pb2.SplitInfo()128 split_info.ParseFromString(splits[0])129 left_child = split_info.left_child.vector130 right_child = split_info.right_child.vector131 split_node = split_info.split_node.categorical_id_binary_split157 split_info = split_info_pb2.SplitInfo()158 split_info.ParseFromString(splits[1])159 left_child = split_info.left_child.vector160 right_child = split_info.right_child.vector161 split_node = split_info.split_node.categorical_id_binary_split[all …]
141 split_info = split_info_pb2.SplitInfo()142 split_info.ParseFromString(splits[0])143 left_child = split_info.left_child.vector144 right_child = split_info.right_child.vector145 split_node = split_info.split_node.dense_float_binary_split170 split_info = split_info_pb2.SplitInfo()171 split_info.ParseFromString(splits[1])172 left_child = split_info.left_child.vector173 right_child = split_info.right_child.vector174 split_node = split_info.split_node.dense_float_binary_split[all …]
48 learner::SplitInfo split_info; member66 if (TF_PREDICT_FALSE(split->split_info.split_node().node_case() == in UpdateBestSplit()82 if (!(IsLeafWellFormed(split->split_info.right_child()) && in UpdateBestSplit()83 IsLeafWellFormed(split->split_info.left_child()))) { in UpdateBestSplit()100 << "(current split: " << split->split_info.split_node().node_case() in UpdateBestSplit()102 << best_split.split_info.split_node().node_case() << ")"; in UpdateBestSplit()103 if (split->split_info.split_node().node_case() < in UpdateBestSplit()104 best_split.split_info.split_node().node_case()) { in UpdateBestSplit()106 } else if (split->split_info.split_node().node_case() == in UpdateBestSplit()107 best_split.split_info.split_node().node_case()) { in UpdateBestSplit()[all …]
284 SplitInfo split_info; in ComputeNormalDecisionTree() local286 split_info.mutable_split_node()->mutable_dense_float_binary_split(); in ComputeNormalDecisionTree()291 auto* left_child = split_info.mutable_left_child(); in ComputeNormalDecisionTree()292 auto* right_child = split_info.mutable_right_child(); in ComputeNormalDecisionTree()296 split_info.SerializeToString(&(*output_splits)(root_idx)); in ComputeNormalDecisionTree()653 SplitInfo split_info; in Compute() local656 dense_split = split_info.mutable_split_node() in Compute()660 dense_split = split_info.mutable_split_node() in Compute()673 auto* left_child = split_info.mutable_left_child(); in Compute()674 auto* right_child = split_info.mutable_right_child(); in Compute()[all …]
27 srcs = ["split_info.proto"],
3 tensorflow/contrib/boosted_trees/proto/split_info.proto
1935 *split_info; in SplitStringInfo() local1941 split_info=AcquireStringInfo(offset); in SplitStringInfo()1942 SetStringInfo(split_info,string_info); in SplitStringInfo()1946 return(split_info); in SplitStringInfo()
1062 partition_ids, gains, split_info) = handler.make_splits(1067 split_info_list.append(split_info)