Add back most of the proto
This commit is contained in:
parent
de3de808b6
commit
251e710762
@ -106,6 +106,29 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
trees {
|
||||||
|
nodes {
|
||||||
|
bucketized_split {
|
||||||
|
feature_id: 75
|
||||||
|
threshold: 21
|
||||||
|
left_id: 1
|
||||||
|
right_id: 2
|
||||||
|
}
|
||||||
|
metadata {
|
||||||
|
gain: -1.4
|
||||||
|
}
|
||||||
|
}
|
||||||
|
nodes {
|
||||||
|
leaf {
|
||||||
|
scalar: -0.6
|
||||||
|
}
|
||||||
|
}
|
||||||
|
nodes {
|
||||||
|
leaf {
|
||||||
|
scalar: 0.165
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
""", ensemble_proto)
|
""", ensemble_proto)
|
||||||
ensemble = boosted_trees_ops.TreeEnsemble(
|
ensemble = boosted_trees_ops.TreeEnsemble(
|
||||||
'ensemble',
|
'ensemble',
|
||||||
|
Loading…
Reference in New Issue
Block a user