| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| tensor_forest_test.py | 33 num_trees=100,
47 num_trees=100,
56 num_trees=100,
71 num_trees=10,
87 num_trees=10,
103 num_trees=10,
123 num_trees=10,
146 num_trees=10,
|
| tensor_forest.py | 89 proto.num_trees = params.num_trees
137 # hparams = ForestHParams(num_classes=5, num_trees=10, num_features=5).fill()
145 num_trees=100,
168 self.num_trees = num_trees
210 self.bagged_num_features) for _ in range(self.num_trees)]
344 for i in range(params.num_trees):
348 for i in range(params.num_trees):
380 for i in range(self.params.num_trees)
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| estimator.py | 35 num_trees=None,
52 num_trees: An int, number of trees to build.
96 'num_trees': num_trees,
114 num_trees=None,
133 num_trees: An int, number of trees to build.
163 'num_trees': num_trees,
184 num_trees=None,
200 num_trees: An int, number of trees to build
[all...] |
| model.py | 62 num_trees = params["num_trees"]
117 if num_trees:
119 num_trees += 1
122 trainer_hooks.StopAfterNTrees(num_trees, attempted_trees,
|
| dnn_tree_combined_estimator.py | 64 dnn_feature_columns, tree_learner_config, num_trees,
83 num_trees: Number of trees to grow model to after training DNN.
245 num_trees += 1
252 num_trees, attempted_trees, finalized_trees)])
264 num_trees,
288 num_trees: Number of trees to grow model to after training DNN.
336 tree_learner_config, num_trees, tree_examples_per_layer, config,
354 num_trees,
377 num_trees: Number of trees to grow model to after training DNN.
430 tree_learner_config, num_trees, tree_examples_per_layer, config
[all...] |
| dnn_tree_combined_estimator_test.py | 75 num_trees=1,
91 num_trees=1,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/models/ |
| forest_to_data_then_nn.py | 42 for i in range(self.params.num_trees)],
45 self.params.num_trees,
|
| 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))
|
| forest_to_data_then_nn_test.py | 42 num_trees=3,
|
| k_feature_decisions_to_data_then_nn_test.py | 42 num_trees=17,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/resources/ |
| decision_tree_ensemble_resource.h | 47 int32 num_trees() const { return decision_tree_ensemble_->trees_size(); } function in class:tensorflow::boosted_trees::models::DecisionTreeEnsembleResource
69 const int num_trees = decision_tree_ensemble_->trees_size(); local
70 if (num_trees <= 0 || LastTreeMetadata()->is_finalized()) {
147 QCHECK_LT(index, num_trees());
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| random_forest_test.py | 34 num_trees=3,
53 num_trees=3,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| training_ops.cc | 254 const int32 num_trees = ensemble_resource->num_trees(); local
255 if (num_trees <= 0) {
265 } else if (num_trees == 1) {
477 const int32 num_trees = ensemble_resource->num_trees(); local
485 const int32 current_tree = num_trees - 1;
503 std::vector<int32> increment_num_updates(num_trees, 0);
509 for (int i = 0; i < num_trees; ++i) {
522 const auto num_trees = ensemble_resource->num_trees() local
529 << num_trees + 1 << " trees."; local
772 const int num_trees = ensemble_resource->num_trees(); variable
[all...] |
| prediction_ops.cc | 65 const std::vector<int32>& trees_to_drop, const int32 num_trees,
68 trees_to_include->reserve(num_trees - trees_to_drop.size());
72 for (int32 tree = 0; tree < num_trees; ++tree) {
247 trees_not_to_drop.insert(ensemble_resource->num_trees() - 1);
359 (ensemble_resource->num_trees() <= 0 ||
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| dropout_utils.cc | 59 const auto num_trees = weights.size(); local
85 for (int32 i = 0; i < num_trees; ++i) {
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/ |
| hybrid_layer_test.py | 37 num_trees=17,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
| binary_mnist.py | 24 --num_eval_steps=1 --num_trees=10 --l2=1 --vmodule=training_ops=1
93 num_trees=FLAGS.num_trees,
162 "--num_trees",
|
| boston.py | 25 --num_eval_steps=1 --num_trees=500 --l2=0.001 \
72 num_trees=FLAGS.num_trees,
162 "--num_trees",
|
| boston_combined.py | 23 --num_trees=100 --tree_l2=0.001 --num_eval_steps=1 \
67 num_trees=FLAGS.num_trees,
158 "--num_trees",
|
| mnist.py | 25 --num_trees=10 --l2=1 --vmodule=training_ops=1
95 num_trees=FLAGS.num_trees,
164 "--num_trees",
|
| /external/tensorflow/tensorflow/examples/learn/ |
| random_forest_mnist.py | 42 num_trees=FLAGS.num_trees,
119 '--num_trees',
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/layers/ |
| decisions_to_data_test.py | 41 num_trees=17,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| training_ops_test.py | 191 self.assertEqual(stats.num_trees, 0)
240 self.assertEqual(stats.num_trees, 0)
266 self.assertEqual(stats.num_trees, 1)
377 self.assertEqual(stats.num_trees, 1)
566 self.assertEqual(stats.num_trees, 0)
748 self.assertEqual(stats.num_trees, 2)
809 self.assertEqual(stats.num_trees, 0)
921 self.assertEqual(stats.num_trees, 1)
[all...] |
| prediction_ops_test.py | 743 num_trees = 10
748 for i in range(0, num_trees):
788 self.assertEqual(num_dropped, num_trees)
790 self.assertEqual(num_dropped_center, num_trees - 1)
800 self.assertEqual(num_trees - 1, dropout_info[0][num_dropped - 1])
805 self.assertEqual(num_trees - 1, dropout_info_center[0][num_dropped_center
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| gbdt_batch.py | 59 NUM_TREES_ATTEMPTED = "num_trees"
[all...] |
