Home
last modified time | relevance | path

Searched refs:num_quantiles (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Destimator.py59 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 …]
Dmodel.py85 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/
Dquantile_ops_test.py42 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/
Dquantile_ops_test.py84 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/
Dquantile_stream_resource.h34 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/
Dweighted_quantiles_summary.h265 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()
Dweighted_quantiles_stream.h147 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/
Dweighted_quantiles_summary.h257 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()
Dweighted_quantiles_stream.h147 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/
Dordinal_split_handler.py99 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,
Dordinal_split_handler_test.py74 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/
Dquantile_ops.py95 num_quantiles, argument
114 self._num_quantiles = num_quantiles
142 num_quantiles=self._num_quantiles,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc187 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/
Dgbdt_batch.py308 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/
Dquantiles.proto11 int64 num_quantiles = 2; field
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc73 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/
Dapi_def_BoostedTreesQuantileStreamResourceFlush.pbtxt19 bool; If True, the output will be the num_quantiles for each stream where the ith
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py102 num_quantiles, argument
107 self._num_quantiles = num_quantiles