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

Try to repro with a single test

Browse Source
This commit is contained in:
Mihai Maruseac 2020-12-30 15:10:11 -08:00
parent ebda2124b5
commit 98ef15b713
1 changed files with 3 additions and 120 deletions

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

@ -39,127 +39,10 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
ensemble = boosted_trees_ops.TreeEnsemble('ensemble') ensemble = boosted_trees_ops.TreeEnsemble('ensemble')
resources.initialize_resources(resources.shared_resources()).run() resources.initialize_resources(resources.shared_resources()).run()
_p("testCreate 2", file=sys.stderr) _p("testCreate 2", file=sys.stderr)
@test_util.run_deprecated_v1
def testCreateWithProto(self):
_p("testCreateWithProto 1", file=sys.stderr)
with self.cached_session(): with self.cached_session():
_p("testCreateWithProto 2", file=sys.stderr) ensemble = boosted_trees_ops.TreeEnsemble('ensemble')
ensemble_proto = boosted_trees_pb2.TreeEnsemble() resources.initialize_resources(resources.shared_resources()).run()
_p("testCreateWithProto 3", file=sys.stderr) _p("testCreate 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)
r = resources.shared_resources()
_p("testCreateWithProto 6", file=sys.stderr)
r=resources.initialize_resources(r)
_p("testCreateWithProto 7", file=sys.stderr)
#r.run()
#_p("testCreateWithProto 8", file=sys.stderr)
if __name__ == '__main__': if __name__ == '__main__':