diff options
author | 2018-04-16 18:03:05 -0700 | |
---|---|---|
committer | 2018-04-16 18:06:20 -0700 | |
commit | 1516756b0297b3642689b06128358aeefd67a321 (patch) | |
tree | c06820a9e9c1efbc864d88f897c5d2a6ec15bc94 /tensorflow/core/ops/boosted_trees_ops.cc | |
parent | 7ee54c2f7cdbc7098627a56b4f084f7b6654b662 (diff) |
Adding min node weight regularization
PiperOrigin-RevId: 193131300
Diffstat (limited to 'tensorflow/core/ops/boosted_trees_ops.cc')
-rw-r--r-- | tensorflow/core/ops/boosted_trees_ops.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/ops/boosted_trees_ops.cc b/tensorflow/core/ops/boosted_trees_ops.cc index 4d74e6d63a..88d6eaf819 100644 --- a/tensorflow/core/ops/boosted_trees_ops.cc +++ b/tensorflow/core/ops/boosted_trees_ops.cc @@ -40,6 +40,7 @@ REGISTER_OP("BoostedTreesCalculateBestGainsPerFeature") .Input("l1: float") .Input("l2: float") .Input("tree_complexity: float") + .Input("min_node_weight: float") .Attr("max_splits: int >= 1") .Attr("num_features: int >= 1") // not passed but populated automatically. .Output("node_ids_list: num_features * int32") |