KLDivergence

public class KLDivergence

A metric that computes the Kullback-Leibler divergence loss metric between labels and predictions.

Inherited Constants

Public Constructors

KLDivergence (Ops tf, String name, long seed, Class<T> type)
Creates a KLDivergence metric

Public Methods

Operand <T>
call ( Operand <? extends TNumber > labels, Operand <? extends TNumber > predictions)
Calculates the weighted loss between labels and predictions

Inherited Methods

Public Constructors

public KLDivergence (Ops tf, String name, long seed, Class<T> type)

Creates a KLDivergence metric

Parameters
tf the TensorFlow Ops
name the name of this metric, if null then metric name is getSimpleName() .
seed the seed for random number generation. An initializer created with a given seed will always produce the same random tensor for a given shape and data type.
type the type for the variables and result

Public Methods

public Operand <T> call ( Operand <? extends TNumber > labels, Operand <? extends TNumber > predictions)

Calculates the weighted loss between labels and predictions

Parameters
labels the truth values or labels
predictions the predictions
Returns
  • the loss