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

IsotonicRegression

public final class IsotonicRegression

Solves a batch of isotonic regression problems.

Public Methods

static <U extends Number, T extends Number> IsotonicRegression<U>
create(Scope scope, Operand<T> input, Class<U> outputDtype)
Factory method to create a class wrapping a new IsotonicRegression operation.
static <T extends Number> IsotonicRegression<Float>
create(Scope scope, Operand<T> input)
Factory method to create a class wrapping a new IsotonicRegression operation using default output types.
Output<U>
output()
A (batch_size, dim)-tensor holding the per-batch element solutions.
Output<Integer>
segments()
An int32 (batch_size, dim)-tensor with the segments.

Inherited Methods

Public Methods

public static IsotonicRegression<U> create (Scope scope, Operand<T> input, Class<U> outputDtype)

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

Parameters
scope current scope
input A (batch_size, dim)-tensor holding a batch of inputs.
outputDtype Dtype of output.
Returns
  • a new instance of IsotonicRegression

public static IsotonicRegression<Float> create (Scope scope, Operand<T> input)

Factory method to create a class wrapping a new IsotonicRegression operation using default output types.

Parameters
scope current scope
input A (batch_size, dim)-tensor holding a batch of inputs.
Returns
  • a new instance of IsotonicRegression

public Output<U> output ()

A (batch_size, dim)-tensor holding the per-batch element solutions.

public Output<Integer> segments ()

An int32 (batch_size, dim)-tensor with the segments.