Home
last modified time | relevance | path

Searched refs:tree_ensemble_config (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dprediction_ops_test.py206 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
210 tree_ensemble_config=tree_ensemble_config.SerializeToString(),
228 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
229 tree = tree_ensemble_config.trees.add()
230 tree_ensemble_config.tree_metadata.add().is_finalized = True
233 tree_ensemble_config.tree_weights.append(1.0)
237 tree_ensemble_config=tree_ensemble_config.SerializeToString(),
256 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
257 tree = tree_ensemble_config.trees.add()
258 tree_ensemble_config.tree_metadata.add().is_finalized = True
[all …]
Dmodel_ops_test.py102 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
103 tree = tree_ensemble_config.trees.add()
105 tree_ensemble_config.tree_weights.append(1.0)
113 tree_ensemble_config=tree_ensemble_config.SerializeToString(),
137 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
139 tree1 = tree_ensemble_config.trees.add()
142 tree_ensemble_config.tree_weights.append(1.0)
145 tree2 = tree_ensemble_config.trees.add()
146 tree_ensemble_config.tree_weights.append(1.0)
158 tree_ensemble_config=tree_ensemble_config.SerializeToString(),
[all …]
Dtraining_ops_test.py155 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
158 tree_ensemble_config=tree_ensemble_config.SerializeToString(),
194 tree_ensemble_config.ParseFromString(serialized)
222 self.assertProtoEquals(expected_result, tree_ensemble_config)
243 tree_ensemble_config.ParseFromString(serialized)
271 self.assertProtoEquals(expected_result, tree_ensemble_config)
306 tree_ensemble_config = tree_config_pb2.DecisionTreeEnsembleConfig()
309 tree_ensemble_config=tree_ensemble_config.SerializeToString(),
362 tree_ensemble_config.ParseFromString(serialized)
410 self.assertProtoEquals(expected_result, tree_ensemble_config)
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees_test.cc57 DecisionTreeEnsembleConfig tree_ensemble_config; in TEST_F() local
64 MultipleAdditiveTrees::Predict(tree_ensemble_config, {}, batch_features_, in TEST_F()
73 DecisionTreeEnsembleConfig tree_ensemble_config; in TEST_F() local
74 auto* tree1 = tree_ensemble_config.add_trees(); in TEST_F()
78 auto* tree2 = tree_ensemble_config.add_trees(); in TEST_F()
91 tree_ensemble_config.add_tree_weights(1.0); in TEST_F()
92 tree_ensemble_config.add_tree_weights(1.0); in TEST_F()
102 MultipleAdditiveTrees::Predict(tree_ensemble_config, {0, 1}, in TEST_F()
114 MultipleAdditiveTrees::Predict(tree_ensemble_config, {0, 1}, in TEST_F()
130 DecisionTreeEnsembleConfig weighted = tree_ensemble_config; in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dprediction_ops_test.py69 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
103 """, tree_ensemble_config)
107 'ensemble', serialized_proto=tree_ensemble_config.SerializeToString())
136 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
173 """, tree_ensemble_config)
177 'ensemble', serialized_proto=tree_ensemble_config.SerializeToString())
209 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
284 """, tree_ensemble_config)
288 'ensemble', serialized_proto=tree_ensemble_config.SerializeToString())
324 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
[all …]
Dtraining_ops_test.py191 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
224 """, tree_ensemble_config)
228 'ensemble', serialized_proto=tree_ensemble_config.SerializeToString())
376 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
419 """, tree_ensemble_config)
423 'ensemble', serialized_proto=tree_ensemble_config.SerializeToString())
526 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
559 """, tree_ensemble_config)
563 'ensemble', serialized_proto=tree_ensemble_config.SerializeToString())
683 tree_ensemble_config = boosted_trees_pb2.TreeEnsemble()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dmodel_ops.py85 tree_ensemble_config=restored_tensors[1])
91 def __init__(self, stamp_token, tree_ensemble_config, name, container=None): argument
93 self._tree_ensemble_config = tree_ensemble_config
127 tree_ensemble_config, argument
143 tree_ensemble_var = TreeEnsembleVariable(stamp_token, tree_ensemble_config,
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch_test.py218 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
321 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
478 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
585 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
690 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
762 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
826 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
910 tree_ensemble_config=ensemble_config.SerializeToString(),
1013 tree_ensemble_config=ensemble_config.SerializeToString(),
1056 stamp_token=0, tree_ensemble_config="", name="tree_ensemble")
[all …]
Dgbdt_batch.py574 local_ensemble_handle, stamp_token=-1, tree_ensemble_config="")
591 tree_ensemble_config=serialized_model), ensemble_stamp
965 tree_ensemble_config="",
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dmodel.py112 tree_ensemble_config="", # Initialize an empty ensemble.
270 tree_ensemble_config="", # Initialize an empty ensemble.
Ddnn_tree_combined_estimator.py249 tree_ensemble_config="", # Initialize an empty ensemble.