| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | estimator.py | 47 num_trees=None, argument
133 'num_trees': num_trees,
155 num_trees=None, argument
226 'num_trees': num_trees,
251 num_trees=None, argument
308 'num_trees': num_trees,
331 num_trees=None, argument
400 'num_trees': num_trees,
427 num_trees=None, argument
503 'num_trees': num_trees,
[all …]
|
| D | dnn_tree_combined_estimator.py | 74 num_trees, argument
321 num_trees += 1
350 trainer_hooks.StopAfterNTrees(num_trees, attempted_trees,
381 trainer_hooks.StopAfterNTrees(num_trees, attempted_trees,
397 num_trees, argument
494 num_trees=num_trees,
524 num_trees, argument
625 num_trees=num_trees,
658 num_trees, argument
741 num_trees=num_trees,
[all …]
|
| D | estimator_test.py | 153 num_trees=1,
172 num_trees=1,
199 num_trees=1,
219 num_trees=1,
239 num_trees=1,
263 num_trees=1,
287 num_trees=1,
307 num_trees=1,
331 num_trees=1,
357 num_trees=1,
[all …]
|
| D | model.py | 80 num_trees = params["num_trees"]
146 if num_trees:
148 num_trees += 1
152 trainer_hooks.StopAfterNTrees(num_trees, attempted_trees,
243 num_trees = params["num_trees"]
364 if num_trees:
366 num_trees += 1
371 trainer_hooks.StopAfterNTrees(num_trees, attempted_trees,
|
| D | dnn_tree_combined_estimator_test.py | 83 num_trees=1,
99 num_trees=1,
122 num_trees=1,
157 num_trees=1,
188 num_trees=1,
217 num_trees=1,
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | resource_ops_test.py | 40 (_, num_trees, num_finalized_trees, num_attempted_layers,
42 self.assertEqual(0, self.evaluate(num_trees))
157 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
160 self.assertEqual(2, self.evaluate(num_trees))
171 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
174 self.assertEqual(0, self.evaluate(num_trees))
223 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
226 self.assertEqual(1, self.evaluate(num_trees))
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| D | tensor_forest_test.py | 37 num_trees=100,
51 num_trees=100,
60 num_trees=100,
75 num_trees=10,
91 num_trees=10,
107 num_trees=10,
125 num_trees=1,
168 num_trees=10,
191 num_trees=10,
|
| D | tensor_forest.py | 90 proto.num_trees = params.num_trees
146 num_trees=100, argument
169 self.num_trees = num_trees
211 self.bagged_num_features) for _ in range(self.num_trees)]
346 for i in range(params.num_trees):
350 for i in range(params.num_trees):
391 for i in range(self.params.num_trees)
435 trees_per_trainer = self.params.num_trees / num_trainers
503 for i in range(self.params.num_trees):
523 self.params.num_trees,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resource_ops.cc | 88 const int num_trees = tree_ensemble_resource->num_trees(); in Compute() local
90 (num_trees <= 0 || in Compute()
91 tree_ensemble_resource->IsTreeFinalized(num_trees - 1)) in Compute()
92 ? num_trees in Compute()
93 : num_trees - 1; in Compute()
118 output_num_trees_t->scalar<int32>()() = num_trees; in Compute()
|
| D | training_ops.cc | 164 if (ensemble_resource->num_trees() > 0) { in Compute()
178 int32 num_trees = ensemble_resource->num_trees(); in UpdateGlobalAttemptsAndRetrieveGrowableTree() local
179 int32 current_tree = num_trees - 1; in UpdateGlobalAttemptsAndRetrieveGrowableTree()
187 if (num_trees <= 0) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()
287 if (ensemble_resource->num_trees() == 0) { in Compute()
|
| D | prediction_ops.cc | 104 const int32 latest_tree = resource->num_trees() - 1; in Compute()
222 if (resource->num_trees() <= 0) { in Compute()
227 const int32 last_tree = resource->num_trees() - 1; in Compute()
329 const int32 last_tree = resource->num_trees() - 1; in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 261 const int32 num_trees = ensemble_resource->num_trees(); in RetrieveBias() local
262 if (num_trees <= 0) { in RetrieveBias()
272 } else if (num_trees == 1) { in RetrieveBias()
571 const int32 num_trees = ensemble_resource->num_trees(); in UpdateTreeWeightsIfDropout() local
579 const int32 current_tree = num_trees - 1; in UpdateTreeWeightsIfDropout()
597 std::vector<int32> increment_num_updates(num_trees, 0); in UpdateTreeWeightsIfDropout()
603 for (int i = 0; i < num_trees; ++i) { in UpdateTreeWeightsIfDropout()
617 const auto num_trees = ensemble_resource->num_trees(); in UpdateAndRetrieveGrowableTree() local
618 if (num_trees <= 0 || in UpdateAndRetrieveGrowableTree()
623 VLOG(1) << "Adding layer #0 to tree #" << num_trees << " of ensemble of " in UpdateAndRetrieveGrowableTree()
[all …]
|
| D | prediction_ops.cc | 66 const std::vector<int32>& trees_to_drop, const int32 num_trees, in CalculateTreesToInclude() argument
69 trees_to_include->reserve(num_trees - trees_to_drop.size()); in CalculateTreesToInclude()
73 for (int32 tree = 0; tree < num_trees; ++tree) { in CalculateTreesToInclude()
255 trees_not_to_drop.insert(ensemble_resource->num_trees() - 1); in DoCompute()
283 {batch_size, ensemble_resource->num_trees()}, in DoCompute()
397 (ensemble_resource->num_trees() <= 0 || in DoCompute()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| D | random_forest_test.py | 66 num_trees=3,
89 num_trees=5,
110 num_trees=1,
149 num_trees=100,
175 num_trees=3,
203 num_trees=5,
228 num_trees=3,
255 num_trees=3,
274 num_trees=5,
296 num_trees=1,
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/models/ |
| D | decisions_to_data_then_nn_test.py | 42 num_trees=17,
59 self.assertEquals(self.params.num_trees, 17)
80 isinstance(self.params.num_trees, tensor_forest.ForestHParams))
89 isinstance(self.params.num_trees, tensor_forest.ForestHParams))
|
| D | forest_to_data_then_nn.py | 42 for i in range(self.params.num_trees)],
45 self.params.num_trees,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/resources/ |
| D | decision_tree_ensemble_resource.h | 47 int32 num_trees() const { return decision_tree_ensemble_->trees_size(); } in num_trees() function
69 const int num_trees = decision_tree_ensemble_->trees_size(); in IncrementAttempts() local
70 if (num_trees <= 0 || LastTreeMetadata()->is_finalized()) { in IncrementAttempts()
148 QCHECK_LT(index, num_trees()); in SetTreeWeight()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | training_ops_test.py | 216 self.assertEqual(stats.num_trees, 0)
265 self.assertEqual(stats.num_trees, 0)
291 self.assertEqual(stats.num_trees, 1)
404 self.assertEqual(stats.num_trees, 1)
517 self.assertEqual(stats.num_trees, 1)
708 self.assertEqual(stats.num_trees, 0)
891 self.assertEqual(stats.num_trees, 2)
953 self.assertEqual(stats.num_trees, 0)
1066 self.assertEqual(stats.num_trees, 1)
1171 self.assertEqual(stats.num_trees, 0)
[all …]
|
| D | prediction_ops_test.py | 825 num_trees = 10
830 for i in range(0, num_trees):
870 self.assertEqual(num_dropped, num_trees)
872 self.assertEqual(num_dropped_center, num_trees - 1)
882 self.assertEqual(num_trees - 1, dropout_info[0][num_dropped - 1])
887 self.assertEqual(num_trees - 1, dropout_info_center[0][num_dropped_center
893 num_trees = 10
898 for i in range(0, num_trees):
905 tree_ensemble_config.growing_metadata.num_trees_attempted = num_trees
906 tree_ensemble_config.growing_metadata.num_layers_attempted = num_trees
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | dropout_utils.cc | 59 const auto num_trees = weights.size(); in DropOutTrees() local
85 for (int32 i = 0; i < num_trees; ++i) { in DropOutTrees()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
| D | mnist.py | 95 num_trees=FLAGS.num_trees,
|
| D | binary_mnist.py | 93 num_trees=FLAGS.num_trees,
|
| D | boston_combined.py | 67 num_trees=FLAGS.num_trees,
|
| D | boston.py | 73 num_trees=FLAGS.num_trees,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/ |
| D | hybrid_layer_test.py | 37 num_trees=17,
|