ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

BoostedTreesBucketize

public final class BoostedTreesBucketize

Bucketize each feature based on bucket boundaries.

An op that returns a list of float tensors, where each tensor represents the bucketized values for a single feature.

Public Methods

List<Output<Integer>>
buckets()
int; List of Rank 1 Tensors each containing the bucketized values for a single feature.
static BoostedTreesBucketize
create(Scope scope, Iterable<Operand<Float>> floatValues, Iterable<Operand<Float>> bucketBoundaries)
Factory method to create a class wrapping a new BoostedTreesBucketize operation.
Iterator<Operand<Integer>>

Inherited Methods

Public Methods

public List<Output<Integer>> buckets ()

int; List of Rank 1 Tensors each containing the bucketized values for a single feature.

public static BoostedTreesBucketize create (Scope scope, Iterable<Operand<Float>> floatValues, Iterable<Operand<Float>> bucketBoundaries)

Factory method to create a class wrapping a new BoostedTreesBucketize operation.

Parameters
scope current scope
floatValues float; List of Rank 1 Tensor each containing float values for a single feature.
bucketBoundaries float; List of Rank 1 Tensors each containing the bucket boundaries for a single feature.
Returns
  • a new instance of BoostedTreesBucketize

public Iterator<Operand<Integer>> iterator ()