tft.bucketize_per_key
Stay organized with collections
Save and categorize content based on your preferences.
Returns a bucketized column, with a bucket index assigned to each input.
tft.bucketize_per_key(
x: common_types.ConsistentTensorType,
key: common_types.ConsistentTensorType,
num_buckets: int,
epsilon: Optional[float] = None,
weights: Optional[common_types.ConsistentTensorType] = None,
name: Optional[str] = None
) -> common_types.ConsistentTensorType
Args |
x
|
A numeric input Tensor , SparseTensor , or RaggedTensor with rank 1,
whose values should be mapped to buckets. CompositeTensor s will have
their non-missing values mapped and missing values left as missing.
|
key
|
A Tensor , SparseTensor , or RaggedTensor with the same shape as
x and dtype tf.string. If x is a CompositeTensor , key must
exactly match x in everything except values, i.e. indices and
dense_shape or nested row splits must be identical.
|
num_buckets
|
Values in the input x are divided into approximately
equal-sized buckets, where the number of buckets is num_buckets.
|
epsilon
|
(Optional) see bucketize .
|
weights
|
(Optional) A Tensor , SparseTensor , or RaggedTensor with the
same shape as x and dtype tf.float32. Used as weights for quantiles
calculation. If x is a CompositeTensor , weights must exactly match
x in everything except values.
|
name
|
(Optional) A name for this operation.
|
Returns |
A Tensor , SparseTensor , or RaggedTensor of the same shape as x , with
each element in the returned tensor representing the bucketized value.
Bucketized value is in the range [0, actual_num_buckets). If the computed
key vocabulary doesn't have an entry for key then the resulting bucket is
-1.
|
Raises |
ValueError
|
If value of num_buckets is not > 1.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-04-26 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-04-26 UTC."],[],[]]