Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator_test.py147 learner_config.constraints.max_tree_depth = 1
166 learner_config.constraints.max_tree_depth = 1
188 learner_config.constraints.max_tree_depth = 1
213 learner_config.constraints.max_tree_depth = 1
233 learner_config.constraints.max_tree_depth = 1
253 learner_config.constraints.max_tree_depth = 1
281 learner_config.constraints.max_tree_depth = 2
301 learner_config.constraints.max_tree_depth = 2
321 learner_config.constraints.max_tree_depth = 1
347 learner_config.constraints.max_tree_depth = 1
[all …]
Ddnn_tree_combined_estimator_test.py91 learner_config.constraints.max_tree_depth = 1
114 learner_config.constraints.max_tree_depth = 1
148 learner_config.constraints.max_tree_depth = 3
179 learner_config.constraints.max_tree_depth = 3
208 learner_config.constraints.max_tree_depth = 3
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/
Dtraining_ops_test.py54 config.constraints.max_tree_depth = max_depth
352 max_tree_depth=learner_config.constraints.max_tree_depth,
466 max_tree_depth=learner_config.constraints.max_tree_depth,
618 max_tree_depth=learner_config.constraints.max_tree_depth,
804 max_tree_depth=learner_config.constraints.max_tree_depth,
942 max_tree_depth=learner_config.constraints.max_tree_depth,
1014 max_tree_depth=learner_config.constraints.max_tree_depth,
1117 max_tree_depth=learner_config.constraints.max_tree_depth,
1200 max_tree_depth=learner_config.constraints.max_tree_depth,
1268 max_tree_depth=learner_config.constraints.max_tree_depth,
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch_test.py224 learner_config.constraints.max_tree_depth = 1
327 learner_config.constraints.max_tree_depth = 2
484 learner_config.constraints.max_tree_depth = 1
591 learner_config.constraints.max_tree_depth = 1
696 learner_config.constraints.max_tree_depth = 1
768 learner_config.constraints.max_tree_depth = 1
832 learner_config.constraints.max_tree_depth = 1
918 learner_config.constraints.max_tree_depth = 1
1021 learner_config.constraints.max_tree_depth = 1
1067 learner_config.constraints.max_tree_depth = 1
[all …]
Dgbdt_batch.py404 if learner_config.constraints.max_tree_depth == 0:
406 learner_config.constraints.max_tree_depth = 6
417 initial_value=self._learner_config.constraints.max_tree_depth)
1119 max_tree_depth=self._max_tree_depth,
1135 max_tree_depth=self._max_tree_depth,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc399 const int32 max_tree_depth = max_tree_depth_t->scalar<int32>()(); in Compute() local
405 dropout_seed, max_tree_depth, in Compute()
616 const int32 max_tree_depth, const int32 weak_learner_type) { in UpdateAndRetrieveGrowableTree() argument
628 tree_metadata->set_is_finalized(max_tree_depth <= 1); in UpdateAndRetrieveGrowableTree()
639 tree_metadata->set_is_finalized(new_num_layers >= max_tree_depth); in UpdateAndRetrieveGrowableTree()
/external/tensorflow/tensorflow/contrib/boosted_trees/examples/
Dmnist.py80 learner_config.constraints.max_tree_depth = FLAGS.depth
Dbinary_mnist.py82 learner_config.constraints.max_tree_depth = FLAGS.depth
Dboston_combined.py58 learner_config.constraints.max_tree_depth = FLAGS.tree_depth
Dboston.py60 learner_config.constraints.max_tree_depth = FLAGS.depth
/external/tensorflow/tensorflow/contrib/boosted_trees/proto/
Dlearner.proto21 uint32 max_tree_depth = 1; field