/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
D | estimator.py | 59 num_quantiles=100): argument 141 'num_quantiles': num_quantiles, 167 num_quantiles=100): argument 234 'num_quantiles': num_quantiles, 262 num_quantiles=100): argument 316 'num_quantiles': num_quantiles, 343 num_quantiles=100): argument 409 'num_quantiles': num_quantiles, 438 num_quantiles=100): argument 511 'num_quantiles': num_quantiles, [all …]
|
D | model.py | 85 num_quantiles = params["num_quantiles"] 128 num_quantiles=num_quantiles) 249 num_quantiles = params["num_quantiles"] 313 num_quantiles=num_quantiles)
|
/external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
D | quantile_ops_test.py | 42 def _gen_config(self, eps, num_quantiles): argument 45 config.num_quantiles = num_quantiles 109 num_quantiles = 3 112 init_stamp_token=0, num_quantiles=num_quantiles, 136 self.assertEqual(num_quantiles + 1, len(buckets)) 142 num_quantiles = 3 147 init_stamp_token=0, num_quantiles=num_quantiles, 183 self.assertEqual(num_quantiles + 1, len(buckets)) 187 self, inputs, num_quantiles=3, expected_buckets=None): argument 194 init_stamp_token=0, num_quantiles=num_quantiles, [all …]
|
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
D | quantile_ops_test.py | 84 self.num_quantiles = constant_op.constant(3, dtype=dtypes.int64) 97 quantile_accumulator_handle, self.num_quantiles) 127 quantile_accumulator_handle_0, self.num_quantiles) 129 quantile_accumulator_handle_1, self.num_quantiles) 150 num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") 169 num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") 182 num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0") 200 num_streams=2, num_quantiles=3, epsilon=self.eps, name="q0")
|
/external/tensorflow/tensorflow/contrib/boosted_trees/resources/ |
D | quantile_stream_resource.h | 34 QuantileStreamResource(const float epsilon, const int32 num_quantiles, in QuantileStreamResource() argument 40 num_quantiles_(num_quantiles), in QuantileStreamResource() 67 int32 num_quantiles() const { return num_quantiles_; } in num_quantiles() function
|
/external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
D | weighted_quantiles_summary.h | 265 std::vector<ValueType> GenerateQuantiles(int64 num_quantiles) const { in GenerateQuantiles() argument 270 num_quantiles = std::max(num_quantiles, int64{2}); in GenerateQuantiles() 271 output.reserve(num_quantiles + 1); in GenerateQuantiles() 275 for (size_t cur_idx = 0, rank = 0; rank <= num_quantiles; ++rank) { in GenerateQuantiles() 278 WeightType d_2 = 2 * (rank * entries_.back().max_rank / num_quantiles); in GenerateQuantiles()
|
D | weighted_quantiles_stream.h | 147 std::vector<ValueType> GenerateQuantiles(int64 num_quantiles) const { in GenerateQuantiles() argument 151 return local_summary_.GenerateQuantiles(num_quantiles); in GenerateQuantiles()
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/quantiles/ |
D | weighted_quantiles_summary.h | 257 std::vector<ValueType> GenerateQuantiles(int64 num_quantiles) const { in GenerateQuantiles() argument 262 num_quantiles = std::max(num_quantiles, int64{2}); in GenerateQuantiles() 263 output.reserve(num_quantiles + 1); in GenerateQuantiles() 267 for (size_t cur_idx = 0, rank = 0; rank <= num_quantiles; ++rank) { in GenerateQuantiles() 270 WeightType d_2 = 2 * (rank * entries_.back().max_rank / num_quantiles); in GenerateQuantiles()
|
D | weighted_quantiles_stream.h | 147 std::vector<ValueType> GenerateQuantiles(int64 num_quantiles) const { in GenerateQuantiles() argument 151 return local_summary_.GenerateQuantiles(num_quantiles); in GenerateQuantiles()
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/ |
D | ordinal_split_handler.py | 99 num_quantiles, argument 149 num_quantiles=num_quantiles, 169 num_quantiles, argument 207 num_quantiles=num_quantiles, 352 num_quantiles, argument 388 num_quantiles=num_quantiles,
|
D | ordinal_split_handler_test.py | 74 num_quantiles=10, 208 num_quantiles=10, 344 num_quantiles=10, 486 num_quantiles=3, 574 num_quantiles=3, 658 num_quantiles=10, 733 num_quantiles=10, 867 num_quantiles=10, 978 num_quantiles=2, 1101 num_quantiles=2, [all …]
|
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
D | quantile_ops.py | 95 num_quantiles, argument 114 self._num_quantiles = num_quantiles 142 num_quantiles=self._num_quantiles,
|
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
D | quantile_ops.cc | 187 int num_quantiles) { in GenerateQuantiles() argument 190 std::vector<float> boundaries = stream.GenerateQuantiles(num_quantiles); in GenerateQuantiles() 191 CHECK_EQ(boundaries.size(), num_quantiles + 1); in GenerateQuantiles() 625 ? GenerateQuantiles(*stream, streams_resource->num_quantiles()) in Compute() 626 : GenerateBoundaries(*stream, streams_resource->num_quantiles())); in Compute() 823 stream, sparse_configs_[sparse_index].num_quantiles()); in Compute() 839 stream, dense_configs_[dense_index].num_quantiles()); in Compute()
|
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
D | gbdt_batch.py | 308 num_quantiles=100): argument 415 self._num_quantiles = num_quantiles 725 num_quantiles = self._num_quantiles 726 epsilon = 1.0 / num_quantiles 741 num_quantiles=num_quantiles, 765 num_quantiles=num_quantiles,
|
/external/tensorflow/tensorflow/contrib/boosted_trees/proto/ |
D | quantiles.proto | 11 int64 num_quantiles = 2; field
|
/external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
D | quantile_ops.cc | 73 const int64 num_quantiles) { in GenerateQuantiles() argument 76 std::vector<float> boundaries = stream.GenerateQuantiles(num_quantiles - 1); in GenerateQuantiles() 77 CHECK_EQ(boundaries.size(), num_quantiles); in GenerateQuantiles()
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_BoostedTreesQuantileStreamResourceFlush.pbtxt | 19 bool; If True, the output will be the num_quantiles for each stream where the ith
|
/external/tensorflow/tensorflow/python/ops/ |
D | boosted_trees_ops.py | 102 num_quantiles, argument 107 self._num_quantiles = num_quantiles
|