experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

More removal

Browse Source
This commit is contained in:
Mihai Maruseac 2020-12-23 09:52:11 -08:00
parent 860d5c79d2
commit a841427991
1 changed files with 1 additions and 127 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

128
tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
View File

@ -50,133 +50,7 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
@test_util.run_deprecated_v1
def testCreateWithProto(self):
_p("testCreateWithProto 1", file=sys.stderr)
with self.cached_session():
_p("testCreateWithProto 2", file=sys.stderr)
ensemble_proto = boosted_trees_pb2.TreeEnsemble()
_p("testCreateWithProto 3", file=sys.stderr)
text_format.Merge(
"""
trees {
nodes {
bucketized_split {
feature_id: 4
left_id: 1
right_id: 2
}
metadata {
gain: 7.62
}
}
nodes {
bucketized_split {
threshold: 21
left_id: 3
right_id: 4
}
metadata {
gain: 1.4
original_leaf {
scalar: 7.14
}
}
}
nodes {
bucketized_split {
feature_id: 1
threshold: 7
left_id: 5
right_id: 6
}
metadata {
gain: 2.7
original_leaf {
scalar: -4.375
}
}
}
nodes {
leaf {
scalar: 6.54
}
}
nodes {
leaf {
scalar: 7.305
}
}
nodes {
leaf {
scalar: -4.525
}
}
nodes {
leaf {
scalar: -4.145
}
}
}
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
}
}
}
tree_weights: 0.15
tree_weights: 1.0
tree_metadata {
num_layers_grown: 2
is_finalized: true
}
tree_metadata {
num_layers_grown: 1
is_finalized: false
}
growing_metadata {
num_trees_attempted: 2
num_layers_attempted: 6
last_layer_node_start: 16
last_layer_node_end: 19
}
""", ensemble_proto)
_p("testCreateWithProto 4", file=sys.stderr)
ensemble = boosted_trees_ops.TreeEnsemble(
'ensemble',
stamp_token=7,
serialized_proto=ensemble_proto.SerializeToString())
_p("testCreateWithProto 5", file=sys.stderr)
resources.initialize_resources(resources.shared_resources()).run()
_p("testCreateWithProto 6", file=sys.stderr)
(stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
nodes_range) = ensemble.get_states()
_p("testCreateWithProto 7", file=sys.stderr)
self.assertEqual(7, self.evaluate(stamp_token))
_p("testCreateWithProto 8", file=sys.stderr)
self.assertEqual(2, self.evaluate(num_trees))
_p("testCreateWithProto 9", file=sys.stderr)
self.assertEqual(1, self.evaluate(num_finalized_trees))
_p("testCreateWithProto 10", file=sys.stderr)
self.assertEqual(6, self.evaluate(num_attempted_layers))
_p("testCreateWithProto 11", file=sys.stderr)
self.assertAllEqual([16, 19], self.evaluate(nodes_range))
_p("testCreateWithProto 12", file=sys.stderr)
pass
if __name__ == '__main__':