| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | estimator_test.py | 145 learner_config = learner_pb2.LearnerConfig()
146 learner_config.num_classes = 2
147 learner_config.constraints.max_tree_depth = 1
152 learner_config=learner_config,
164 learner_config = learner_pb2.LearnerConfig()
165 learner_config.num_classes = 2
166 learner_config.constraints.max_tree_depth = 1
171 learner_config=learner_config,
186 learner_config = learner_pb2.LearnerConfig()
187 learner_config.num_classes = 2
[all …]
|
| D | dnn_tree_combined_estimator_test.py | 73 learner_config = learner_pb2.LearnerConfig()
74 learner_config.num_classes = 2
82 tree_learner_config=learner_config,
89 learner_config = learner_pb2.LearnerConfig()
90 learner_config.num_classes = 2
91 learner_config.constraints.max_tree_depth = 1
98 tree_learner_config=learner_config,
112 learner_config = learner_pb2.LearnerConfig()
113 learner_config.num_classes = 2
114 learner_config.constraints.max_tree_depth = 1
[all …]
|
| D | estimator.py | 44 learner_config, argument
122 if learner_config.num_classes == 0:
123 learner_config.num_classes = n_classes
124 elif learner_config.num_classes != n_classes:
126 (learner_config.num_classes, n_classes))
132 'learner_config': learner_config,
152 learner_config, argument
217 learner_config.num_classes = 2
219 learner_config.num_classes = label_dimension
225 'learner_config': learner_config,
[all …]
|
| D | model.py | 76 learner_config = params["learner_config"]
122 learner_config=learner_config,
239 learner_config = params["learner_config"]
307 learner_config=learner_config,
329 learner_config=learner_config,
|
| D | dnn_tree_combined_estimator.py | 262 learner_config=tree_learner_config,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/ |
| D | node-stats_test.cc | 32 LearnerConfig learner_config; in TEST() local
33 learner_config.set_multi_class_strategy(LearnerConfig::TREE_PER_CLASS); in TEST()
34 NodeStats node_stats(learner_config, GradientStats(1e-8f, 1e-8f)); in TEST()
40 LearnerConfig learner_config; in TEST() local
41 learner_config.set_multi_class_strategy(LearnerConfig::TREE_PER_CLASS); in TEST()
42 learner_config.mutable_constraints()->set_min_node_weight(3.2f); in TEST()
43 NodeStats node_stats(learner_config, GradientStats(7.32f, 1.63f)); in TEST()
49 LearnerConfig learner_config; in TEST() local
50 learner_config.set_multi_class_strategy(LearnerConfig::TREE_PER_CLASS); in TEST()
51 learner_config.mutable_regularization()->set_l1(10.0f); in TEST()
[all …]
|
| D | split-stats.h | 38 SplitStats(const LearnerConfig& learner_config, const NodeStats& root_stats) in SplitStats()
46 SplitStats(const LearnerConfig& learner_config, const NodeStats& root_stats, in SplitStats()
52 learner_config.regularization().tree_complexity()) {} in SplitStats()
|
| D | node-stats.h | 50 NodeStats(const LearnerConfig& learner_config, in NodeStats()
52 : NodeStats(learner_config.regularization().l1(), in NodeStats()
53 learner_config.regularization().l2(), in NodeStats()
54 learner_config.constraints().min_node_weight(), in NodeStats()
55 learner_config.multi_class_strategy(), grad_stats) {} in NodeStats()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch_test.py | 219 learner_config = learner_pb2.LearnerConfig()
220 learner_config.learning_rate_tuner.fixed.learning_rate = 0.1
221 learner_config.num_classes = 2
222 learner_config.regularization.l1 = 0
223 learner_config.regularization.l2 = 0
224 learner_config.constraints.max_tree_depth = 1
225 learner_config.constraints.min_node_weight = 0
235 learner_config=learner_config,
322 learner_config = learner_pb2.LearnerConfig()
323 learner_config.num_classes = 2
[all …]
|
| D | gbdt_batch.py | 300 learner_config, argument
339 if learner_config is None:
342 if learner_config.num_classes < 2:
360 if (learner_config.multi_class_strategy ==
363 learner_config.multi_class_strategy = (
366 learner_config.multi_class_strategy = (
369 if logits_dimension == 1 or learner_config.multi_class_strategy == (
378 if (learner_config.multi_class_strategy ==
385 if (learner_config.growing_mode ==
387 learner_config.growing_mode = learner_pb2.LearnerConfig.LAYER_BY_LAYER
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| D | prediction_ops_test.py | 165 learner_config, argument
178 learner_config=learner_config,
186 learner_config, argument
197 learner_config=learner_config,
215 learner_config = learner_pb2.LearnerConfig()
216 learner_config.num_classes = 2
220 learner_config=learner_config.SerializeToString(),
242 learner_config = learner_pb2.LearnerConfig()
243 learner_config.num_classes = 2
247 learner_config=learner_config.SerializeToString(),
[all …]
|
| D | training_ops_test.py | 163 learner_config = _gen_learner_config(
183 learner_config=learner_config)
234 learner_config=learner_config)
281 learner_config=learner_config)
314 learner_config = _gen_learner_config(
349 learner_config=learner_config.SerializeToString(),
352 max_tree_depth=learner_config.constraints.max_tree_depth,
424 learner_config = _gen_learner_config(
463 learner_config=learner_config.SerializeToString(),
466 max_tree_depth=learner_config.constraints.max_tree_depth,
[all …]
|
| D | model_ops_test.py | 108 learner_config = learner_pb2.LearnerConfig()
109 learner_config.num_classes = 2
125 learner_config=learner_config.SerializeToString(),
175 learner_config = learner_pb2.LearnerConfig()
176 learner_config.num_classes = 3
186 learner_config=learner_config.SerializeToString(),
216 learner_config = learner_pb2.LearnerConfig()
217 learner_config.num_classes = 2
252 learner_config=learner_config.SerializeToString(),
281 learner_config=learner_config.SerializeToString(),
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
| D | mnist.py | 72 learner_config = learner_pb2.LearnerConfig()
76 learner_config.learning_rate_tuner.fixed.learning_rate = FLAGS.learning_rate
77 learner_config.num_classes = num_classes
78 learner_config.regularization.l1 = 0.0
79 learner_config.regularization.l2 = FLAGS.l2 / FLAGS.examples_per_layer
80 learner_config.constraints.max_tree_depth = FLAGS.depth
83 learner_config.growing_mode = growing_mode
86 learner_config.multi_class_strategy = (
91 learner_config=learner_config,
|
| D | binary_mnist.py | 77 learner_config = learner_pb2.LearnerConfig()
79 learner_config.learning_rate_tuner.fixed.learning_rate = FLAGS.learning_rate
80 learner_config.regularization.l1 = 0.0
81 learner_config.regularization.l2 = FLAGS.l2 / FLAGS.examples_per_layer
82 learner_config.constraints.max_tree_depth = FLAGS.depth
85 learner_config.growing_mode = growing_mode
90 learner_config=learner_config,
|
| D | boston.py | 56 learner_config = learner_pb2.LearnerConfig()
57 learner_config.learning_rate_tuner.fixed.learning_rate = FLAGS.learning_rate
58 learner_config.regularization.l1 = 0.0
59 learner_config.regularization.l2 = FLAGS.l2
60 learner_config.constraints.max_tree_depth = FLAGS.depth
66 learner_config=learner_config,
|
| D | boston_combined.py | 53 learner_config = learner_pb2.LearnerConfig()
54 learner_config.learning_rate_tuner.fixed.learning_rate = (
56 learner_config.regularization.l1 = 0.0
57 learner_config.regularization.l2 = FLAGS.tree_l2
58 learner_config.constraints.max_tree_depth = FLAGS.tree_depth
66 tree_learner_config=learner_config,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
| D | prediction_ops.cc | 30 LearnerConfig learner_config; in ApplyGradientTreesPredictionShapeFn() local
31 ParseProtoUnlimited(&learner_config, learner_config_str); in ApplyGradientTreesPredictionShapeFn()
37 reduce_dim ? learner_config.num_classes() - 1 in ApplyGradientTreesPredictionShapeFn()
38 : learner_config.num_classes())}); in ApplyGradientTreesPredictionShapeFn()
46 LearnerConfig learner_config; in ApplyGradientTreesPredictionVerboseShapeFn() local
47 ParseProtoUnlimited(&learner_config, learner_config_str); in ApplyGradientTreesPredictionVerboseShapeFn()
53 reduce_dim ? learner_config.num_classes() - 1 in ApplyGradientTreesPredictionVerboseShapeFn()
54 : learner_config.num_classes())}); in ApplyGradientTreesPredictionVerboseShapeFn()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | prediction_ops.cc | 103 LearnerConfig learner_config; in GradientTreesPredictionOp() local
108 context, ParseProtoUnlimited(&learner_config, learner_config_str), in GradientTreesPredictionOp()
111 num_classes_ = learner_config.num_classes(); in GradientTreesPredictionOp()
115 context, ParseProtoUnlimited(&learner_config, learner_config_str), in GradientTreesPredictionOp()
123 learner_config.growing_mode() == learner_config.WHOLE_TREE; in GradientTreesPredictionOp()
124 if (learner_config.has_learning_rate_tuner() && in GradientTreesPredictionOp()
125 learner_config.learning_rate_tuner().tuner_case() == in GradientTreesPredictionOp()
127 dropout_config_ = learner_config.learning_rate_tuner().dropout(); in GradientTreesPredictionOp()
136 apply_averaging_ && learner_config.averaging_config().config_case() != in GradientTreesPredictionOp()
139 averaging_config_ = learner_config.averaging_config(); in GradientTreesPredictionOp()
|
| D | training_ops.cc | 62 const boosted_trees::learner::LearnerConfig& learner_config, in UpdateBestSplit() argument
74 if (learner_config.pruning_mode() == in UpdateBestSplit()
|